fixed off by one bug
Merge branch 'master' of https://github.com/johnnyg12791/cs224n