fix the last equation of qm-hw-1