diff --git a/src/PythonBridge/PBApplication.class.st b/src/PythonBridge/PBApplication.class.st
index ee5b9ae..4a0aeff 100644
--- a/src/PythonBridge/PBApplication.class.st
+++ b/src/PythonBridge/PBApplication.class.st
@@ -44,6 +44,11 @@ PBApplication class >> do: aBlockClosure [
 	^ retVal
 ]
 
+{ #category : #initialization }
+PBApplication class >> initialize [
+	SessionManager default registerToolClassNamed: self name
+]
+
 { #category : #testing }
 PBApplication class >> isRunning [
 	^ uniqueInstance notNil and: [ uniqueInstance isRunning ]
@@ -88,6 +93,11 @@ PBApplication class >> send: obj [
 	^ self uniqueInstance send: obj
 ]
 
+{ #category : #accessing }
+PBApplication class >> shutdown: isImageQuitting [
+	isImageQuitting ifTrue: [ self stop ]
+]
+
 { #category : #'start-stop' }
 PBApplication class >> start [
 	self isRunning ifTrue: [ Error signal: 'Keras already running.' ].