rewrite highlight source block method