fix something in lab 4 homework, lab 5