Add .onreturn