New XAI Code (Fork)