I updated the code to highlight the principle a bit better.