Added comments to the notebook