updated models to better our project