This is great! Would it make sense to add this to the ```basic_agent_usage``` and ```shell``` layouts as well?