Edit createBall method