Fix code display issues