Решил, но явно можно было проще с теме же if else. Подскажите пожалуйста?