adding categories, hw part 2