Update digit.py