Edited the notebook for clarity

Added more markdown text and some headers