Updated map code