updated models slightly