Add finishAfter functionality.