Add wait_for_completion parameter to Snapshot.restore_snapshot()