Add websockets and chat