Style fixes

Went through the entire codebase to make sure everything's with PEP 8