Adding test for non-local return