I would prefer the ability to auto paint the train based on contents rather than on destination station colour (I know there is a mode for this) but otherwise all excellent features.

Instead of just being a yellow circle the colour could change based on the upgrade.
For example if upgrading from a yellow belt to a red belt. The left half of the icon could be yellow and the right half be red.

This may help in identifying what an item is being upgrade to.