update lab2 framework