Slight change to master.py