Improve mappings