Centering withen a JPanel

[RKhadder]

This is my code

public class FaciltyLogIn extends JPanel implements ActionListener
{
    public static JPasswordField codeInput;
    public static JDialog dialog;
    public static boolean loggedIn = false;
    public static String user = "";
    
    public FaciltyLogIn()
    {
        if (!loggedIn)
        {
            JLabel label = new JLabel("Password");
            codeInput = new JPasswordField(20);
            JButton button = new JButton("OK");
            button.setActionCommand("OK");
            button.addActionListener(this);
        
            add(label);
            add(codeInput);
            add(button);
        }
        else if (loggedIn)
        {
            Calendar calendar = Calendar.getInstance();
            SimpleDateFormat dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
            JLabel timeLabel = new JLabel("You timed in at " + dateFormat.format(calendar.getTime()));
            JLabel nameLabel = new JLabel("Welcome back " + user);
            JButton button = new JButton("CLOSE");
            button.setActionCommand("CLOSE");
            button.addActionListener(this);
            
            add(timeLabel);
            add(nameLabel);
            add(button);
        }
    }

   ...

   private static void GUI()
    {
        dialog = new JDialog();
        
        FaciltyLogIn contentPane = new FaciltyLogIn();
        contentPane.setOpaque(true);
        dialog.setContentPane(contentPane);
        
        Dimension size = Toolkit.getDefaultToolkit().getScreenSize();
        size.setSize(size.getWidth(), size.getHeight());
        
        dialog.toFront();
        dialog.setSize(size);
        dialog.setDefaultCloseOperation(JDialog.DO_NOTHING_ON_CLOSE);
        dialog.setResizable(false);
        dialog.setUndecorated(true);
        dialog.setAlwaysOnTop(true);
        dialog.setVisible(true);
    }

Now skip to the second if-statment. As you see it adds the two labels and the button. However when they are added they are right next to each other. How do I make it so they are right after each other and in the center of dialog? I tried adding the .setLayout(new GridLayout(0,1)) but it uses the whole screen so that doesnt work.

[masijade]

Try GridBagLayout

[RKhadder]

That work nicely thank you very very much.

GridBagLayout gridBag = new GridBagLayout();
            GridBagConstraints constraints = new GridBagConstraints();
            setLayout(gridBag);
            constraints.fill = GridBagConstraints.BOTH;
            constraints.weightx = 1.0;
            constraints.gridwidth = GridBagConstraints.REMAINDER;
            gridBag.setConstraints(timeLabel, constraints);
            constraints.gridwidth = GridBagConstraints.REMAINDER;
            gridBag.setConstraints(nameLabel, constraints);
            constraints.gridwidth = GridBagConstraints.REMAINDER;
            gridBag.setConstraints(button, constraints);

However, the button spans the whole width, from oneside of the screen to the other.