Added working upgrade screen