Added the capacity to change the colour of the balls when selected