add proper database models