updated repository to work with EF