не сломалось

Как добавить условие if, чтобы ничего не сломалось, или по другому решить эту задачу?.