woocommerce function

Hello, I need a woocommerce function that I will add with code snippets plugin to my woocommerce store to hide sub category products on category listing.
At the moment woocommerce lists category and sub category products.

For example I have this category tree

Main Category (0 products)
--- Sub Category (2 products)
------Sub Sub Category (5 products)

If I go to Sub category it displays 7 products.

I wish to display:

- on Main Category only the Sub Category
- on Sub Category the Sub Sub Category plus Sub Category 2 products
- on Sub Sub Category the 5 products

I have already tried the following code but it hides the Sub Category on Main category. Sub Category should not be hidden

https://www.templatemonster.com/help/woocommerce-how-to-exclude-subcategory-products-from-category-listing-page.html