lecture 3 fix