Подскажите пожалуйста почему не подходит решение? По мне так самый простой способ это через if() проверить. Код кстати работает