PEP8 fixes pt 2