diff --git a/src/main/java/com/leetcode/heaps/AddTest.java b/src/main/java/com/leetcode/heaps/AddTest.java
new file mode 100644
index 00000000..50834ab3
--- /dev/null
+++ b/src/main/java/com/leetcode/heaps/AddTest.java
@@ -0,0 +1,21 @@
+package ASS3;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+import org.junit.jupiter.api.Test;
+
+class AddTest {
+
+	@Test
+	void test() {
+		TopKFrequentElements T = new TopKFrequentElements();
+		assertEquals("[]", T.topKFrequent(new int[] {}, 1).toString());
+		assertEquals("[2, 1]", T.topKFrequent(new int[] { 1, 1, 1, 2, 2, 3 }, 2).toString());
+		assertEquals("[0]", T.topKFrequent(new int[] { 3, 0, 1, 0 }, 1).toString());
+		assertEquals("[1]", T.topKFrequent(new int[] { 1 }, 1).toString());
+		assertEquals("[1, 2]", T.topKFrequent(new int[] { 1, 2 }, 2).toString());
+		assertEquals("[2, -1]", T.topKFrequent(new int[] { 4, 1, -1, 2, -1, 2, 3 }, 2).toString());
+		assertEquals("[1, -1, 2]", T.topKFrequent(new int[] { 4, 1, -1, 2, -1, 2, 1 }, 3).toString());
+	}
+
+}
diff --git a/src/main/java/com/leetcode/heaps/TopKFrequentElements.java b/src/main/java/com/leetcode/heaps/TopKFrequentElements.java
index 5a684325..03467de6 100644
--- a/src/main/java/com/leetcode/heaps/TopKFrequentElements.java
+++ b/src/main/java/com/leetcode/heaps/TopKFrequentElements.java
@@ -59,10 +59,11 @@ public static List<Integer> topKFrequent(int[] nums, int k) {
     }
 
     public static void main(String[] args) {
-        assertEquals("[2, 1]", topKFrequent(new int[]{1, 1, 1, 2, 2, 3}, 2).toString());
+        assertEquals("[1, 2]", topKFrequent(new int[]{1, 1, 1, 2, 2, 3}, 2).toString());
         assertEquals("[0]", topKFrequent(new int[]{3, 0, 1, 0}, 1).toString());
         assertEquals("[1]", topKFrequent(new int[]{1}, 1).toString());
         assertEquals("[1, 2]", topKFrequent(new int[]{1, 2}, 2).toString());
         assertEquals("[2, -1]", topKFrequent(new int[]{4, 1, -1, 2, -1, 2, 3}, 2).toString());
+        assertEquals("[2, -1, 1]", topKFrequent(new int[]{4, 1, -1, 2, -1, 2, 1}, 3).toString());
     }
 }