Updated the Upgrade Screen to include the right upgrade classes.