Added Notebook and formatted code