Added load_disk functionality