making flake8 happy
removing unused imports
removing unused and shouldn't be working function
Merge pull request #1 from prometheanfire/master

cleaning up the code