Improved test connection UI