Add the imports I forgot earlier
Merge pull request #36 from planetguy32/master

Add the imports I forgot earlier