fix lab 4