resolve compile error.

I assumed that lecture 3 code should be the same as lecture 2.
Merge pull request #2 from sosxisang/patch-2

resolve compile error.