Basic style fixes

PEP8 & stuff.