Property Hive - add view shortlist button to results page

snippet.php

add_action('propertyhive_before_search_results_loop', 'add_view_shortlist_button', 50);
function add_view_shortlist_button() {
    if ( isset($_GET['shortlisted']) && $_GET['shortlisted'] == '1' ) {
        echo '<div class="view-shortlist-container" style="clear: both;">
                <a href="/search/" class="button view-shortlist-button">Back to Search</a>
              </div>';
    }
    echo '<div class="view-shortlist-container" style="clear: both;">
            <a href="/search/?shortlisted=1" class="button view-shortlist-button">View Shortlist</a>
          </div>';
}