Fixed the problems with the ball reference