Simulation code for a magneto optical trap