added __init__ for module
Removed useless import