use traitlet for highlight_class