homework 5 bug fix?