Changed type of array in Map, added JUnit Test file