adding to lecture 2