PEP8 cleanup

Compulsive formatting!