Add data context and repository