How to show stacked categories on mobile category pages - squarespace 7.1?

Site URL: https://oval-lute-tdat.squarespace.com/

I would like to get my categories visible and stack like this (image attached) when I am on my shop page and shop category pages on mobile. These stacked categories are only showing on /shop and not /shop/category. I would also like it to show the "all collections" category which isn't showing up for some reason with the code I was using to center/wrap the categories. 

Any tips?

Code used in Custom CSS below:

/*Stacking shop categories on mobile*/
//Option #1
@media screen and (max-width: 575px) {
  .products.collection-content-wrapper .nested-category-children {
    flex-wrap: wrap;
    justify-content: center;
    overflow-x: hidden;
    padding-top: 15px;
  }
}

Hi, It looks like you figured it out?