Update lab 2 solution