Update app/models/square.rb