Fixed it I think

Indeed
Merge pull request #7 from JeremyGee/master

Fixed it I think!