Small refactor

Хочу сделать небольшие изменения