Рефакторинг

Сделал небольшой рефакторинг существующего кода