Add Lab2 original code