Use break instead of return