cleaned the notebook, and added comments