Modified migration tool to use a map in order to simplify migrations