I think changing it to end after the ball disappears is a good idea.