Fixed bugs HW6