Generalized upgrade screen code

Also started working on station upgrades