Adding unit of work for EF.