Сделал небольшой рефакторинг