fix the jquery to change class