fix code highlight issues