Re-add missing functions.php stuff