Updated get_disk_usage implementation