clean up display of code