ML assignment 3 - predict hand-written digits using trained neural network