I update my modified code for lab-2.