Немного рефакторинг.