Add search tests