add missing statement to add class to span