refactored: Переработан код (немного)