Add code for noise fitting (ARIMA model)