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