Невеличкий рефакторинг.