@Override
public void actionPerformed(ActionEvent e) {
// If figure icon matches the the button selected's icon - insert figure into board
Icon icon = ((JButton) e.getSource()).getIcon();
if (icon.equals(rook.getFigureIcon())) {
location.setFigure(rook);
} else if (icon.equals(knight.getFigureIcon())) {
location.setFigure(knight);
} else if (icon.equals(bishop.getFigureIcon())) {
location.setFigure(bishop);
} else if (icon.equals(queen.getFigureIcon())) {
location.setFigure(queen);
}
// Delete frame when clicked
dispose();
}
