Update homework_2_2_solution.py