Update problem22.py