Update diskcheck.py