Update Assignment2.rb