Changed the ball to a square