Updated for HW6