Update problem 0009.py