Added more headings to the .ipynb