Import missing class
Merge pull request #4 from macnibblet/patch-1

Import missing class