Update exercise_2.1