Cleaned up code.
Made upgrades disable themselves, when they would normally crash the game.