Fixed crash and updated upgrade screen.