Update Assignment2.rb

Beautification