Further tweaks to `total_delay` handler arg

- Improve method comment
- Tidy up test
 - Speed up test