Small change to template.php