not following the solution :(