Add repository class