Reformatted code to comply with PEP guidelines