Improve captured page to show code