Added missing use statements

Fixes #134