Implement basic websocket functionality