Added code to get the color of the ball at the beginning.