attempt to fix lab 4