Improve mapper