adding changes to the solution