updated model to run with full sample