add code for ARMA model trading