fix function and class highlighting