Scaleup UI polish

This branch fixes the scale up button colour and sets the unit count from the model.