Fixed include
Merge pull request #2 from bmagyar/master

Little fixes to make it fully compile on ubuntu with cmake