remove unused imports; PEP 8 compliance
Merge pull request #2 from maxcountryman/master

Minor cleanup; PEP8 compliance