axe

Fix race condition in shutdown code (triggered when not doing wait_all)
Fix very rare quasi-race condition in wait code (delayed marking tasks done)
Add new tests

Tested: jam