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