Fixed some PEP8 errors and wrong imports.