digit_recognition.py