add in some code to classify digits