DFT version working!