Bug fixes per HW6