updated models a little bit