Changes to function.php method