This is now implemented in the Revit plugin and the Model Editor. The release notes are here:
Thank you again for the suggestion!