Edited for PEP-8 compliance