Updated to include collision detection between balls