Update QueryExpansion.py