Added upgrade screen