lecture 5 fix