Updated the notebook to include images of data analysis