PEP8 styling fixes