Fixed unused imports warnings
Merge pull request #8 from konradxyz/fix_imports

Fix imports