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