Minor update to master.py