A feature to make it possible to group certain entities in a schema would be nice. Maybe something like the 'System Architecture' example where you could wrap entities in a named block, to group them as a schema?