Updated contract to include IReturn