Added Await as a class