update unit2 model thinking