Got search functionality working