hw6 bug fix