Copper Golem Organization

I think copper golems should have a better system for organizing. Right now, a chest has to contain at least one of the items the golem is trying to organize. Maybe items could be grouped together (wood blocks, farming items, ores, tools, etc.) to help them figure out where to put things. Currently, they do help organize, but if I take out all of one block, they can’t put it back. I would have to put at least one back, then they could sort after. Just feels like an extra step that could be avoided if items were automatically grouped together.