Updated some features and the websockets class