diff --git a/Training_IC_on_Demand.ipynb b/Training_IC_on_Demand.ipynb
index 96829a9..0305c92 100644
--- a/Training_IC_on_Demand.ipynb
+++ b/Training_IC_on_Demand.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "markdown",
-   "id": "7a9ce3d8",
+   "id": "2da7d8b4",
    "metadata": {},
    "source": [
     "# Create ICs on Demand"
@@ -11,7 +11,7 @@
   {
    "cell_type": "code",
    "execution_count": 1,
-   "id": "3b5cc730",
+   "id": "dd080e2e",
    "metadata": {},
    "outputs": [],
    "source": [
@@ -26,7 +26,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "3b4e1ada",
+   "id": "b1dd5f7e",
    "metadata": {},
    "source": [
     "# Setup the model"
@@ -35,7 +35,7 @@
   {
    "cell_type": "code",
    "execution_count": 2,
-   "id": "b521b99e",
+   "id": "47ee081d",
    "metadata": {},
    "outputs": [
     {
@@ -59,7 +59,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "066fb79e",
+   "id": "7bcfec89",
    "metadata": {},
    "source": [
     "# test training with white noise"
@@ -68,7 +68,7 @@
   {
    "cell_type": "code",
    "execution_count": 6,
-   "id": "8001a82b",
+   "id": "1b4e58ae",
    "metadata": {},
    "outputs": [
     {
@@ -100,7 +100,7 @@
   {
    "cell_type": "code",
    "execution_count": 7,
-   "id": "46589304",
+   "id": "71c39acb",
    "metadata": {},
    "outputs": [
     {
@@ -134,7 +134,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "623c9205",
+   "id": "bda66fa1",
    "metadata": {},
    "source": [
     "# load BORG data"
@@ -143,7 +143,7 @@
   {
    "cell_type": "code",
    "execution_count": 8,
-   "id": "725f2c38",
+   "id": "e2405bbe",
    "metadata": {},
    "outputs": [
     {
@@ -168,7 +168,7 @@
   {
    "cell_type": "code",
    "execution_count": 9,
-   "id": "ae7e24a8",
+   "id": "e2d6b3bc",
    "metadata": {},
    "outputs": [
     {
@@ -196,7 +196,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "1abeba05",
+   "id": "8e186b79",
    "metadata": {},
    "source": [
     "## generate sample"
@@ -205,7 +205,7 @@
   {
    "cell_type": "code",
    "execution_count": 10,
-   "id": "8d7b6bf2",
+   "id": "05e0cbbe",
    "metadata": {},
    "outputs": [
     {
@@ -238,7 +238,7 @@
   },
   {
    "cell_type": "markdown",
-   "id": "7b286278",
+   "id": "cac4005d",
    "metadata": {},
    "source": [
     "## get an ensemble"
@@ -247,7 +247,7 @@
   {
    "cell_type": "code",
    "execution_count": 29,
-   "id": "99a8d21a",
+   "id": "de6eb5aa",
    "metadata": {},
    "outputs": [
     {
@@ -283,7 +283,7 @@
   {
    "cell_type": "code",
    "execution_count": 30,
-   "id": "2169349e",
+   "id": "8464f0d7",
    "metadata": {},
    "outputs": [
     {
@@ -323,7 +323,7 @@
   {
    "cell_type": "code",
    "execution_count": null,
-   "id": "7dbea77a",
+   "id": "70af9a01",
    "metadata": {},
    "outputs": [],
    "source": []