Added database models and context