Removed unused imports and separated each import onto a separate line, per PEP8