updates after watching the solution