adding in search for UI