Add connect button functionality