Added changes to staff.py