Improved async/await support