updated models a bit