Updated main.py to include choice of specie for reweighting for tee.