Rewrite the theme toggle method to properly save the settings