Update `echo_done` function