Restore and Spawn units like in legion td

[Xzere]

Hello, i'm trying to make a similar map to legion td, my only issue is how to make the units restore after each wave, so either if they surive or die, the shall be restored. Anyone who have the knowledge to help me out? Thanks in advance.