function wp_demo_shortcode() { 

//Turn on output buffering
ob_start();
$code = 'Hello World';
ob_get_clean();

 // Output needs to be return
return $code;
} 

// register shortcode
add_shortcode('helloworld', 'wp_demo_shortcode');