Add new methods to repository class