edit src/get_disk.py