Improvements to InMemoryDbContext