Added database models and unit of work