Update exercise_3-1.rb