Lab_2 is done
Fix bugs in lab_2