Made code display more accurate