diff --git a/notebooks/wut-train.ipynb b/notebooks/wut-train.ipynb
index 57916f9..c3269bb 100644
--- a/notebooks/wut-train.ipynb
+++ b/notebooks/wut-train.ipynb
@@ -89,7 +89,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "ENCODING='FM'\n",
+    "ENCODING='FSK9k6'\n",
     "batch_size = 64\n",
     "epochs = 4\n",
     "# Failing with this now:\n",
@@ -125,21 +125,18 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "print('total training good images:', num_train_good)\n",
-    "print('total training bad images:', num_train_bad)\n",
-    "#print(\"--\")\n",
-    "print(\"Total training images:\", total_train)\n",
-    "print('total validation good images:', num_val_good)\n",
-    "print('total validation bad images:', num_val_bad)\n",
-    "print(\"--\")\n",
-    "print(\"Total validation images:\", total_val)\n",
-    "#print(\"Reduce training and validation set when testing\")\n",
-    "#total_train = 100\n",
-    "#total_val = 100\n",
-    "print(\"Train =\")\n",
-    "print(total_train)\n",
-    "print(\"Validation =\")\n",
-    "print(total_val)"
+    "print('Training good images:   ', num_train_good)\n",
+    "print('Training bad images:    ', num_train_bad)\n",
+    "print('Training images:        ', total_train)\n",
+    "print('Validation good images: ', num_val_good)\n",
+    "print('Validation bad images:  ', num_val_bad)\n",
+    "print('Validation images:      ', total_val)\n",
+    "print('')\n",
+    "print('Reduce training and validation set')\n",
+    "total_train = 1000\n",
+    "total_val = 1000\n",
+    "print('Training reduced to:    ', total_train)\n",
+    "print('Validation reduced to:  ', total_val)"
    ]
   },
   {
@@ -323,7 +320,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.summary()"
+    "#model.summary()"
    ]
   },
   {
@@ -444,7 +441,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.save('/srv/satnogs/data/models/FM/wut-FM.h5')"
+    "h5_file=(\"wut-\" + ENCODING + \".h5\")\n",
+    "save_path_h5 = os.path.join('/srv/satnogs/data/models/', ENCODING, h5_file)\n",
+    "print(save_path_h5)"
    ]
   },
   {
@@ -453,7 +452,27 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.save('/srv/satnogs/data/models/FM/wut-FM.tf')"
+    "model.save(save_path_h5)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "tf_modeldir=(\"wut-\" + ENCODING + \".tf\")\n",
+    "save_path_tf = os.path.join('/srv/satnogs/data/models/', ENCODING, tf_modeldir)\n",
+    "print(save_path_tf)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "model.save(save_path_tf)"
    ]
   },
   {