Making it compile

The project didn't quite compile, but with a couple minor changes I got it to.

Thanks for making this available.