fixing highlighting issues for code