Fix PEP8 complaints