Worked on lecture 2