Thank a lot :D
We agree, we thought of having a system where the player is sent back to the wave map directly after selecting the upgrade so they could e.g. drag and drop the upgrade on the tower they want. We just didn't have the time in the end !