Missing imports
Merge pull request #25 from Forever-Young/patch-1

Thanks for your patches!