fix code display bugs