cleaned up PEP 8 stuff