Removed unused imports and fixed User bug via pyflakes