Add storing of norm of test error for neural nets