Minor style update (PEP8)