Formatting compliant with PEP8
Reduced length of problem1