Modified display code