add lab2 codes