We should be able to change our 'remembered' team *after* a battle

The game remembers what pokemon we used last, which is often helpful, but it'd be nice to be able to change the team after. Example: You beat a Dark Grunt with Fighting types, after the battle, there's an option to change the saved team, so you put in Ground because it's better against Muk and the skunks you keep seeing.

You can make your own teams or try to remember and change it next time, but it would be nice if the game didn't keep suggesting the team that just barely won.