WP User Manager - Content Restriction function to restrict parts of a templat

wpum_content_restriction_function.php

<?php
get_header();

if ( wpum_can_access_restricted_content() ) : ?>
  Some extra content in the template that is restricted as per the restriction settings of the page
<?php endif; ?>

<div>
<?php the_content(); ?>
</div>

<?php get_footer(); ?>