updated context to use IDbSet