Fixing Websocket implementation

Currently the web socket doesn't deal with error message received in subscription and also fails when un-subscription is done before the notification for successful subscription has been received.