Fix some pylint errors
Fix bad imports