Particles in python with pygame