added diamond upgrade
added hook for upgrades, called when an upgrade gets added (used by diamond / gold ingot)
removed TestUpgrade class