Добавил возможность выбора соединения
Рефакторинг кода