Update disk_check.py