Remove some unused imports
Remove custom map function on transition maps
Remove transition_map class