<s>[INST] import wx

class MyFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self, None, title="Window Test")
        self.panel = wx.Panel(self)
        self.button = wx.Button(self.panel, label='Close')
        self.button.Bind(wx.EVT_BUTTON, self.on_button)
        self.Show()
        self.button.SetFocus()

    def on_button(self, event):
        self.Close()

if __name__ == "__main__":
    app = wx.App()
    frame = MyFrame()
    app.MainLoop()

In this code, I want to add a choice to the button, so that when the chosen option is selected, it closes the window. How can I do this?[/INST]