Added more upgrade menu stuff