Add lab2 code