JTextField: ввести число и по нажатию кнопки получить его квадрат

package easybudda.examples.squarednumber;
 
import java.awt.*;
import java.awt.event.*;
import javax.swing.*;
 
public class SquaredNumber {
    private JFrame mainFrame;
    private JTextField textField;
    
    SquaredNumber() {
        mainFrame = new JFrame("Squared number");
        mainFrame.setSize(300, 100);
        mainFrame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        mainFrame.setLayout(new FlowLayout());
        
        mainFrame.add(new JLabel("Number:"));
        textField = new JTextField(8);
        mainFrame.add(textField);
        
        JButton button = new JButton("Square it");
        button.addActionListener(new ActionListener() {
            @Override
            public void actionPerformed(ActionEvent ae) {
                try {
                    int i = Integer.parseInt(textField.getText());
                    JOptionPane.showMessageDialog(mainFrame, String.valueOf(i * i), "Squared number", JOptionPane.INFORMATION_MESSAGE);
                }
                catch ( NumberFormatException nfe ) {
                    JOptionPane.showMessageDialog(mainFrame, "Wrong input!", "Error", JOptionPane.ERROR_MESSAGE);
                }
            }
        });
        
        mainFrame.add(button);
        
        mainFrame.pack();
        mainFrame.setVisible(true);
    }
    
    public static void main(String[] args) {
        SwingUtilities.invokeLater(new Runnable() {
            @Override
            public void run() {
                new SquaredNumber();
            }
        });
    }
}