Fixed a bug as per HW6