replace if to Map