Update problemC.py