Updated simulation and added some more documentation to the notebook