Fix imports
Merge pull request #18 from CS673/fixup

Fix imports