Add test to for Map class