Fix build errors

Fixes #2.
Merge pull request #5 from kevva/build

Fix build errors