fix code in lab1.c