updated homework.rb