change to get return vector