Start using WebSockets in the UI