Fixed some errors in notebook, added some code to it