imporve code highlight