Worked on lecture 01