update lesson5 q1 need to clean codes