update codes for Q2 lesson6