Update saetrain.m