Adding more to lecture 2