Add .toggleClass() function