Update dmrg.cpp