Add search function, fix missing IDs in data