added config for which model to run