small change to lecture 3