la3 PartB finished
modify lab3 partb code