Pygame Python3 draws multiple balls bouncing and colliding in a window