fixing class 5 lab 4