Improve API for WebSocketActor