trying to fix the code so the solution prints