Minor changes for PEP8 compliance