Add async functionality