Particle system written in python