Update assign2.rb