Re-add entity framework