Adding WebSockets to the index.js file