Started lab2 code