I added 2 lines of code.

:D