Add test for map()