lab2 Code