added better comments to notebook