add week5A q2 solution code