Scaleup UI polish

This branch fixes the scale up button colour and sets the unit count from the model.
Good idea moving it here instead of duplicating the set up elsewhere