Fix pyling warnings.
FIx PEP8