diff --git a/my-first-dl4j-project/pom.xml b/my-first-dl4j-project/pom.xml
index 88713ebf..18aeae10 100644
--- a/my-first-dl4j-project/pom.xml
+++ b/my-first-dl4j-project/pom.xml
@@ -35,8 +35,8 @@
 
   <properties>
     <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
-    <dl4j.version>1.0.0-beta7</dl4j.version>
-    <nd4j.version>1.0.0-beta7</nd4j.version>
+    <dl4j.version>1.0.0-M1.1</dl4j.version>
+    <nd4j.version>1.0.0-M1.1</nd4j.version>
     <java.version>1.8</java.version>
     <slf4j.version>2.0.0-alpha1</slf4j.version>
     <maven-shade-plugin.version>2.4.3</maven-shade-plugin.version>
@@ -55,14 +55,14 @@
     <!-- to use this, disable nd4j-native-platform and use this -->
     <!-- <dependency> -->
     <!--   <groupId>org.nd4j</groupId> -->
-    <!--   <artifactId>nd4j-cuda-10.2-platform</artifactId> -->
+    <!--   <artifactId>nd4j-cuda-11.2-platform</artifactId> -->
     <!--   <version>${nd4j.version}</version> -->
     <!-- </dependency> -->
 
     <!--futher more enable cudnn when training using gpu -->
     <!-- <dependency> -->
     <!--   <groupId>org.deeplearning4j</groupId> -->
-    <!--   <artifactId>deeplearning4j-cuda-10.2</artifactId> -->
+    <!--   <artifactId>deeplearning4j-cuda-11.2</artifactId> -->
     <!--   <version>${dl4j.version}</version> -->
     <!-- </dependency> -->
 
@@ -73,6 +73,12 @@
       <version>${dl4j.version}</version>
     </dependency>
 
+    <!-- deeplearning4j-ui is used for visualization -->
+    <dependency>
+      <groupId>org.deeplearning4j</groupId>
+      <artifactId>deeplearning4j-ui</artifactId>
+      <version>${dl4j.version}</version>
+    </dependency>
 
     <!-- https://mvnrepository.com/artifact/org.slf4j/slf4j-api -->
     <dependency>
diff --git a/my-first-dl4j-project/src/main/java/ai/certifai/MyFirstDL4JProject.java b/my-first-dl4j-project/src/main/java/ai/certifai/MyFirstDL4JProject.java
index 597152fb..f496560e 100644
--- a/my-first-dl4j-project/src/main/java/ai/certifai/MyFirstDL4JProject.java
+++ b/my-first-dl4j-project/src/main/java/ai/certifai/MyFirstDL4JProject.java
@@ -20,6 +20,7 @@
 import org.datavec.api.records.reader.RecordReader;
 import org.datavec.api.records.reader.impl.csv.CSVRecordReader;
 import org.datavec.api.split.FileSplit;
+import org.deeplearning4j.core.storage.StatsStorage;
 import org.deeplearning4j.datasets.datavec.RecordReaderDataSetIterator;
 import org.deeplearning4j.nn.conf.MultiLayerConfiguration;
 import org.deeplearning4j.nn.conf.NeuralNetConfiguration;
@@ -28,6 +29,9 @@
 import org.deeplearning4j.nn.multilayer.MultiLayerNetwork;
 import org.deeplearning4j.nn.weights.WeightInit;
 import org.deeplearning4j.optimize.listeners.ScoreIterationListener;
+import org.deeplearning4j.ui.api.UIServer;
+import org.deeplearning4j.ui.model.stats.StatsListener;
+import org.deeplearning4j.ui.model.storage.InMemoryStatsStorage;
 import org.nd4j.common.io.ClassPathResource;
 import org.nd4j.linalg.activations.Activation;
 import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;
@@ -86,7 +90,12 @@ public static void main( String[] args ) throws Exception {
         MultiLayerNetwork network = new MultiLayerNetwork(config);
 
         network.init();
-        network.setListeners(new ScoreIterationListener(1));
+
+        // Setup UI and listeners
+        UIServer server = UIServer.getInstance();
+        StatsStorage storage = new InMemoryStatsStorage();
+        server.attach(storage);
+        network.setListeners(new ScoreIterationListener(1), new StatsListener(storage));
 
         network.fit(iterator, 10);