Add whenDone()