Add feature to assign categories
Now you can remove categories