Upgrade shop working but not balanced