Updating to return integer