Made some modifications to lecture 8