Add tests for search and 20k words