Update square.rb