int gridSquares;
do {
String input = JOptionPane.showInputDialog(
"Enter number of rows and columns you want grid to contain\n" +
"(For example entering 2 will create a 2x2 grid)\n");
gridSquares = Integer.parseInt( input );
if(gridSquares < 1 || gridSquares > 5)
JOptionPane.showMessageDialog(null, "Number must be between 1 and 5",
"Input Error", JOptionPane.ERROR_MESSAGE);
} while(gridSquares < 1 || gridSquares > 5);