minor changes, add another category