Fix update from Kelvin Lee
Merge pull request #22 from mrcupid12/kelvinlee93-master

Fix update from Kelvin Lee