Add repository interface and mock class