Add Lab2 code