add return in if