update of code for model comparison