Edit method MyHashMap.put(key, value)

Author: jsquared21

Exercise_27/Exercise_27_02/MyHashMap.class

@@ -90,7 +90,7 @@ public java.util.Set<MyMap.Entry<K, V>> entrySet() {
 	public V get(K key) {
 		int index = hash(key.hashCode());
 		int j = 0;
-		
+
 		while (table.get(index) != null) {
 			if (table.get(index).getKey().equals(key)) {
 				return table.get(index).getValue();
@@ -126,8 +126,8 @@ public V put(K key, V value) {
 		int j = 0;
 
 		while (table.get(index) != null) {
-			// Add a new entry (key, value)
-			if (table.get(index).getKey() == key) {
+			// The key is already in the map
+			if (table.get(index).getKey().equals(key)) {
 				Entry<K, V> entry = table.get(index);
 				V oldValue = entry.getValue();
 				// Replace old value with new value