Edited for PEP8 compliance