Add class to write down the map