Thanks for the suggestion. Feel free to open a PR if you like!