Adding some more of lecture 2