Update nntrain.m