diff --git a/src/main/java/fi/utu/tech/ooj/exercise3/Main.java b/src/main/java/fi/utu/tech/ooj/exercise3/Main.java
index c4fe648c832c7d298d49c7a02091b4f3ad855082..1b2f3bf77bb1c854d308aae56bffd4fef48158ba 100644
--- a/src/main/java/fi/utu/tech/ooj/exercise3/Main.java
+++ b/src/main/java/fi/utu/tech/ooj/exercise3/Main.java
@@ -3,8 +3,9 @@ package fi.utu.tech.ooj.exercise3;
 
 import java.util.HashMap;
 import java.util.HashSet;
+import java.util.Map;
 
-    public class Main {
+public class Main {
 
 
         /**
@@ -26,19 +27,19 @@ import java.util.HashSet;
         HashSet<KuormaAuto> kuormaautot = new HashSet<KuormaAuto>();
         */
             Map<String, String> alkuperäinen = new HashMap<>();
-            originalMap.put("avain1", "arvo1");
-            originalMap.put("avain2", "arvo2");
-            originalMap.put("avain3", "arvo3");
+            alkuperäinen.put("avain1", "arvo1");
+            alkuperäinen.put("avain2", "arvo2");
+            alkuperäinen.put("avain3", "arvo3");
 
-            Map<String, String> randomMap = new RandomMap<>(originalMap);
+            Map<String, String> randomMap = new RandomMap<>(alkuperäinen);
 
-            String value1 = randomMap.get("key1"); // Palauttaa "value1"
-            String value2 = randomMap.get("key4"); // Palauttaa satunnaisen arvon alkuperäisestä arvojoukosta
+            String testi1 = randomMap.get("avain1"); // palauttaa arvo1
+            String testi2 = randomMap.get("avain4"); // Palauttaa satunnaisen arvon
 
-            System.out.println(value1);
-            System.out.println(value2);
+            System.out.println(testi1);
+            System.out.println(testi2);
 
         }
     }
 
-}
+
diff --git a/src/main/java/fi/utu/tech/ooj/exercise3/RandomMap.java b/src/main/java/fi/utu/tech/ooj/exercise3/RandomMap.java
index 8a5ea75dcfb8bf093aae42330c5a2446bf0a0ee8..c8e62b68e03f4f66e677939925263cbda1ebb7b7 100644
--- a/src/main/java/fi/utu/tech/ooj/exercise3/RandomMap.java
+++ b/src/main/java/fi/utu/tech/ooj/exercise3/RandomMap.java
@@ -69,17 +69,17 @@ public class RandomMap<K, V> implements Map<K, V> {
 
     @Override
     public Set<K> keySet() {
-        return null;
+        return AlkuperäinenMap.keySet();
     }
 
     @Override
     public Collection<V> values() {
-        return null;
+         return AlkuperäinenMap.values();
     }
 
     @Override
     public Set<Entry<K, V>> entrySet() {
-        return null;
+        return AlkuperäinenMap.entrySet();
     }