Complete functions.php with an add_action