modified models to be match our data