Fixed final bug.

Results are exactly the same as the reference Piazza post.
Merge branch 'master' of https://github.com/williamleif/cs224n-p1