Working on lab 3 more