First try of adding category input

I’m sure it’s buggy