Updated code to follow PEP 8 Style Guide