Небольшой рефакторинг
Ещё мелкие изменения