Add lab1's code