improve demo page with code hightlight function