Add hightlighting for demo