Rewrite to using wxPython