Feature/database models