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