why `if`? see my example again, I made the same in two times less lines