Please run the the run.py file with a certain index. This index will correspond to the timestep you want to plot. Note that you must choose a timestep which has a high density! The indices with the most data points will be printed out in terminal (for reference)