Improve the UI for adding more categories. Closes #5