cleaned the notebook a bit