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