Better formatting (PEP 8)