Solution correction

Added a correction to the solution.