updating for new models