Working on adding category