Добавил обработку if конструкций.