Change models according to DB