Add websocket handler classes