lab2 code