Extending the solution

7. Extending the solution