almost done with hetero model