Add returning statement support