Add changes for HW6