From 76e513bc89be1b1dad7674482ef7c6eddf9add1b Mon Sep 17 00:00:00 2001
From: Markus Ankenbrand <markus@ankenbrand.me>
Date: Mon, 12 Sep 2016 12:55:49 +0200
Subject: [PATCH] Add header to style

---
 functions.php | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/functions.php b/functions.php
index 37428dd..cfe4451 100644
--- a/functions.php
+++ b/functions.php
@@ -23,31 +23,35 @@ add_filter('acf/fields/google_map/api', 'my_acf_google_map_api');
 
 // Related posts function
 function atravelblog_related_posts() {
-	echo('<div class="related-posts posts section-inner">');
 	global $post;
 	$thisID = get_the_ID();
 	$cat = get_the_category()[0]->name;
 	if ( $cat == 'Urlaub' ){
 		$tagebuch = new WP_Query( array( 'category_name' => 'Tagebuch' ) );
 	    if ($tagebuch->have_posts()) :
-		    echo('<h4>Tagebuch Einträge:</h4>');
+			echo('<div class="related-posts posts section-inner">');
+		    echo('<h4 style="color: white; margin-top: 20px;">Tagebuch Einträge:</h4>');
 	        while ( $tagebuch->have_posts() ) : $tagebuch->the_post();
 		        global $post;
 		        if($thisID == get_field('urlaub')->ID){
 			        get_template_part( 'content', get_post_format() );
 		        }
 	        endwhile;
+		    echo('<div class="clear"></div>');
+		    echo('</div> <!-- /related-posts -->');
         endif;
+		wp_reset_query();
     } elseif($cat == 'Tour') {
         $urlaub = new WP_Query(array('p' => get_field('urlaub')[0]));
 		if ($urlaub->have_posts()) :
+			echo('<div class="related-posts posts section-inner">');
 			while ( $urlaub->have_posts() ) : $urlaub->the_post();
 				global $post;
 				get_template_part( 'content', get_post_format() );
 			endwhile;
+			echo('<div class="clear"></div>');
+			echo('</div> <!-- /related-posts -->');
 		endif;
+		wp_reset_query();
     }
-	echo('<div class="clear"></div>');
-    echo('</div> <!-- /related-posts -->');
-    wp_reset_query();
 }
\ No newline at end of file