Code cleanup. PEP 8 formatting.