Restrict Website to logon Users

function restrict_site_to_logged_in_users() {
if (
!is_user_logged_in() &&
!is_page('login') && // Optional: Replace with your custom login page slug if needed
!is_admin()
) {
wp_redirect(wp_login_url());
exit;
}
}
add_action('template_redirect', 'restrict_site_to_logged_in_users');