Updated for consistency with code in the video
Merge pull request #19 from keefekwan/patch-7

Updated for consistency with code in the video