Added a tiny bit to digit_classification