Update HITGeneration.py