Add better handling of class toggling for #search-form submit button