Add vector search
Not necessary with defaultdict