Modified KPU LAB1 to work every time