Nice sorting puzzle. I cleared all levels. Try to limit moves/swaps to add challenge. And maybe a chubby-type or PWD-type unit that cannot be swapped. 👍
A light calm music, catchy cartoonish sounds, and visual cues would make it livelier and look like a polished mobile game.