tidied up parse tree highlighting