adding category button works now