Can you take this out of the code and put it in an issue?  That's the right place to track future/desired functionality.