561 HW3 bug fixed