minor changes to the solution