Updates to decorators.py