Add more docs for query_entities