Add support for async function timing