Added tox configuration
Removed unused imports
Fixed failing tests