Make WebsocketHandler more abstract