Updated code to match code in video
Merge pull request #21 from keefekwan/patch-9

Updated code to match code in video