Just modified smth
Almost done lab_2_6