Added some comments
Merge pull request #4 from alasdairmackenzie/master

I've added some comments to the code and improved things... honestly.