Updating class on submit button