Updated Hashmap code