use code in lab2