Updated the plotting (reconstructing DM)