Add WIP search functionality