Mod: Clean up code.