UI update for "documents" overview

I have checked with my team, and It seems the Collections can help in your use case.
To read more about setting up collections, you check this article.

Also, we have a similar open feature request that you can check: