Updates to helper.py