input: Write a python function to fit an ARIMA model to a given dataset and make a prediction for the next time step.
  assert arima_prediction([12, 15, 8, 7, 8, 10, 13, 6, 11, 10, 13], (1, 1, 0)) == [12.0]
  assert arima_prediction([20, 25, 30, 35, 40], (2, 1, 1)) == [40.0]
  assert arima_prediction([5, 6, 7, 8, 9], (0, 1, 1)) == [9.0]
output: def arima_prediction(data, order):
      from statsmodels.tsa.arima.model import ARIMA
      
      # Fit the ARIMA model
      model = ARIMA(data, order=order)
      model_fit = model.fit()
      
      # Make prediction for the next time step
      prediction = model_fit.forecast(steps=1)
      
      return prediction.tolist()