Minor model update