-
-from wxPython.lib.PyCrust import PyCrustShell, PyCrustEditor, PyCrustVersion
+from wx import py
 
 
 #----------------------------------------------------------------------
 
-intro = 'Welcome To PyCrust %s - The Flakiest Python Shell' % PyCrustVersion.version
+intro = 'Welcome To PyCrust %s - The Flakiest Python Shell' % py.version.VERSION
 
 def runTest(frame, nb, log):
-    shell = PyCrustShell.Shell(nb, intro)
-    return shell.editor
+    win = py.crust.Crust(nb, intro=intro)
+    return win
 
 #----------------------------------------------------------------------
 
-overview = PyCrustShell.__doc__
+overview = py.filling.__doc__ + "\n\n" + py.crust.__doc__
+
+if __name__ == '__main__':
+    import sys,os
+    import run
+    run.main(['', os.path.basename(sys.argv[0])])
+