Update V1 contracts

Author: ilitteri

claim_contracts/src/AlignedTokenV1.sol

@@ -1,44 +1,27 @@
 // SPDX-License-Identifier: MIT
 pragma solidity ^0.8.19;
 
-import "@openzeppelin/contracts-upgradeable/token/ERC20/ERC20Upgradeable.sol";
-import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
-import "@openzeppelin/contracts-upgradeable/token/ERC20/extensions/ERC20PermitUpgradeable.sol";
 import "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
-import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
+import "@openzeppelin/contracts-upgradeable/token/ERC20/ERC20Upgradeable.sol";
+import "@openzeppelin/contracts/utils/ReentrancyGuard.sol";
 
-contract AlignedTokenV1 is
-    Initializable,
-    ERC20Upgradeable,
-    OwnableUpgradeable,
-    ERC20PermitUpgradeable,
-    UUPSUpgradeable
-{
+contract AlignedTokenV1 is Initializable, ERC20Upgradeable, ReentrancyGuard {
     /// @custom:oz-upgrades-unsafe-allow constructor
     constructor() {
         _disableInitializers();
     }
 
     function initialize(
-        address _initialOwner,
         address _beneficiary1,
         uint256 _beneficiary1Part,
         address _beneficiary2,
         uint256 _beneficiary2Part,
         address _beneficiary3,
         uint256 _beneficiary3Part
-    ) public initializer {
+    ) public initializer nonReentrant {
         __ERC20_init("AlignedTokenV1", "ALI");
-        __Ownable_init(_initialOwner);
-        __ERC20Permit_init("AlignedTokenV1");
-        __UUPSUpgradeable_init();
-
         _mint(_beneficiary1, _beneficiary1Part);
         _mint(_beneficiary2, _beneficiary2Part);
         _mint(_beneficiary3, _beneficiary3Part);
     }
-
-    function _authorizeUpgrade(
-        address newImplementation
-    ) internal override onlyOwner {}
 }

claim_contracts/src/ClaimableAirdropV1.sol

@@ -4,15 +4,15 @@ pragma solidity ^0.8.13;
 import "@openzeppelin/contracts/token/ERC20/IERC20.sol";
 import "@openzeppelin/contracts/utils/cryptography/MerkleProof.sol";
 import "@openzeppelin/contracts/utils/ReentrancyGuard.sol";
-import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
 import "@openzeppelin/contracts-upgradeable/proxy/utils/Initializable.sol";
-import "@openzeppelin/contracts-upgradeable/proxy/utils/UUPSUpgradeable.sol";
+import "@openzeppelin/contracts-upgradeable/utils/PausableUpgradeable.sol";
+import "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
 
 contract ClaimableAirdropV1 is
     ReentrancyGuard,
     Initializable,
-    OwnableUpgradeable,
-    UUPSUpgradeable
+    PausableUpgradeable,
+    OwnableUpgradeable
 {
     address public tokenContractAddress;
     address public tokenOwnerAddress;
@@ -36,8 +36,6 @@ contract ClaimableAirdropV1 is
         bytes32 _claimMerkleRoot
     ) public initializer nonReentrant {
         __Ownable_init(_initialOwner);
-        __UUPSUpgradeable_init();
-
         tokenContractAddress = _tokenContractAddress;
         tokenOwnerAddress = _tokenOwnerAddress;
         limitTimestampToClaim = _limitTimestampToClaim;
@@ -77,7 +75,11 @@ contract ClaimableAirdropV1 is
         emit TokenClaimed(msg.sender, amount);
     }
 
-    function _authorizeUpgrade(
-        address newImplementation
-    ) internal override onlyOwner {}
+    function pause() public onlyOwner {
+        _pause();
+    }
+
+    function unpause() public onlyOwner {
+        _unpause();
+    }
 }