updating to return simply the ID