Python+PyGame elastic collisions simulator