Just a few things I had to change.
Merge pull request #1 from jamesdavidson/small-changes

Just a few things I had to change.