Lecture 2 fix