Update problem010.py