Add stack effect comments to main
Add `with`