Missed some imports
Merge pull request #35 from computmaxer/master

Missed some imports