Updated reassign.py