adding return instead of break