updated comments to problem_2.rb