Added test implementation of await