from galaxysimulation import GalaxyCollisionSimulation sim = GalaxyCollisionSimulation(num_particles=4) sim.simulate(num_steps=10000, time_step=0.001) sim.display_snapshots(200, fix_axes=True)