Working on lab 6