Added await_shutdown() function.