Added more explanation to the notebook