add missing use statements
Merge pull request #4 from prolic/patch-2

add missing use statements