Update graph.py

Rewrite DFT()