update square.rb