Updated homework.rb