Add WebSocket and Mock dependencies