Adding class for repository