Update 2_direct_disks_box.py