updated the solution