Update HITGeneration.py

Now it works