Add disk_free function