fix missing includes (#63)

master has errors right now, fix that

Merged #63.