Yes I had an Idea about letting the user move the units between waves or add new ones as you said... maybe I will add something like this after the jam.
