PeterisP Cleaned up the notebook and comments