Made slight changes to display mode to fix columns and rows