Test connection functionality added