Add improvements compared to the previous version of the model