Yes like that but for change team.
You can not change your team from All Pokemon so I don't usually go there. You can evolve, change attacks, nickname, trade, sell, or release from All Pokemon but not change your team.
That's why I would like the very thing you pointed out available for the Change Team screen.