Updated setSquare inputs