Adding lab1 code