Worked on websocket connection