works now. but still need to return the vector.