Вывести на странице родительской категории ее подкатегории списком и в подкатегориях вывести записи, которые относятся к ним

<?php
    // Получить подкатегории текущей родительской категории
    $categories = get_categories( array(
        'parent' => get_query_var( 'cat' ),
        'hide_empty' => false, 
        'orderby'    => 'ID', 
        'order'      => 'ASC', 
    ) );

    // Если есть подкатегории, то выводим записи внутри них
    if ( $categories ) :
        foreach ( $categories as $category ) :
            $args = array(
                'post_type' => 'post',
                'posts_per_page' => -1,
                'category__in' => array( $category->term_id )
            );
            $query = new WP_Query( $args );

            // Если есть записи, то выводим их внутри подкатегории
            if ( $query->have_posts() ) : ?>
                <h2><?php echo $category->name; ?></h2>
                <ul>
                    <?php while ( $query->have_posts() ) : $query->the_post(); ?>
                        <li><a href="<?php the_permalink(); ?>"><?php the_title(); ?></a></li>
                    <?php endwhile; ?>
                </ul>
            <?php endif;

            // Сбросить запрос для следующей итерации
            wp_reset_postdata();
        endforeach;
    else :
        // Если нет подкатегорий, то выводим записи текущей категории
        if ( have_posts() ) :
            while ( have_posts() ) : the_post(); ?>
                <!-- Здесь может быть HTML-код, если нужно -->
                <h2><?php the_title(); ?></h2>
                <div><?php the_content(); ?></div>
                <!-- Здесь может быть еще HTML-код, если нужно -->
            <?php endwhile;
        endif;
    endif;
?>