Forgot to add some #includes
Merge pull request #162 from dmcdougall/fix-compiler-errors

Forgot to add some #includes