diff --git a/ETA.py b/ETA.py
index 1d9da23..f6b5247 100644
--- a/ETA.py
+++ b/ETA.py
@@ -402,5 +402,8 @@ for x in range( size ):
     for i in range( len( fall ) - 1 ):
         stairs[ fall[ i ] : fall[ i + 1 ] , x ] = np.mean( stairs[ fall[ i ] : fall[ i + 1 ] ] )
     stairs[ fall[ -1 ] : , x ] = 0
-plt.imshow( stairs )
-plt.show()
+
+if output == None:
+    print( stairs )
+else:
+    np.savetxt( output , stairs )