updated hw3.rb