updates to lecture 4