Gravity Simulation (python 3)