How do you change the script so that it keeps the class for the option that was set in the Wordpress menu navigation for the link?