Adding rmse code