Add HW6 fixes