minor changes after watching the solution videos