Cleanup imports
PEP8 cleanup
Merge branch 'cleanup' into release/1.2.2