Remove unnecessary imports from __init__

[jarvisteach]

There are extra imports in the __init__.py file to help with testing, this is not necessary.

Those lines can be moved to the test script.