Yes, the `return` attribute is what should do it. You also need to compile the code with -dip1000 option.