Thanks for this. But its not working.
Adding the code to Function.php is breaking the site.

Any update?