quick change to add model formula (needs to be more robust)