HW 4 Set up