Make code comply to PEP 8