Update prob12.py