HW6 Adding staging