Updated Connection class to handle new WSProtocol class