Small fixes
PEP8 fixes