Merge pull request #9 from srawlins/remove-unused-imports

Remove unused imports in the test files