Updated the IRepository interface