add current user role name to body class in wordpress

<?php

// Add role class to body
function jb_add_role_to_body($classes) {   
    if (is_user_logged_in()) {
        foreach (wp_get_current_user()->roles as $user_role) {
            $classes[] = 'role-'. $user_role;
        }
    }
    return $classes;
}
add_filter('body_class','jb_add_role_to_body');