Add repository class override