changes to lecture 3 and 4