Adding Upgrade support, using WebSocket middleware from sample app