add toy function for DFT