updated notebook for lecture 4