Added test for map