Update lab4 solution