Event driven simulation of hard spheres in python