add return to sub