Update the repository class