This is code for solving the digit classification problem