updated models to reflect db change