Naprawa połączeń i ulepszenie obsługi

Dodałem komenty do lepszego zozumienia kodu.