lab2 code change