Websockets enable on heroku and added to the code again