finish test on training data. 3, 6,7, 9 are easily mistaken with 2, especially 6
add evaluation on test data