Second round of Squares.rb. Made it more ruby-like"