Add get_disk_usage func