Yeah, that's something I definitely love to add in next few days. I plan to use one of the Test Dataset from GraphRAG paper.