Add support for AwaitExpression node