Figured out how to do categories properly