Fix PEP8 warnings and some whitespaces