improve display of code