Small PEP8 corrections