WebSocket integration
Final style dev :)