Added work from class week 4 lecture 1.