Updated Leader connection handling