I added 2 lines of code.

:D
Merge pull request #1 from daliu/patch-1

I added 2 lines of code.