Removing numerical deriv from trainNN