homework 1 part 2 Categories bug fix