Updated wormcalc.py script