updates to lecture 2