Added the entity framework