Reformatted code according to PEP8