More work on category ui