Hi,
OK, mal das offensichtliche vorweg: wenn Du den Code in die functions.php des Themes geschrieben hast, wird es beim nächsten Updates desselben wieder überschrieben, und geht damit verloren. Erstelle besser ein Child Theme, und erzeuge dann dazu eine functions.php . Im Gegensatz zu den übrigen Templates wird die functions.php zusätzlich zu der gleichnamigen Datei abgearbeitet, und nicht anstelle von.
Wenn Du uns einen Link gegeben hättest, könnte man was sagen, ob Dein Vorgehen sinnvoll ist oder weniger. So ist das ein bisschen schwierig zu beurteilen, aber wenn es für Dich so stimmt, wieso nicht.