add more logic for different colors of balls