Update class-frontend.php