minor fixes after watching the solution video