Working on lab 2