Cleanup of the notebook, added comments