Try improving the code display