fixing lab 4