Update comments

src/AavePM.sol

@@ -451,7 +451,7 @@ contract AavePM is
         _storeEventBlockNumber();
     }
 
-    /// @notice // TODO: Add comment and move to different heading. How to protect this function?
+    /// @notice // TODO: Add comment. How to protect this function?
     function delegateCallHelper(string memory _targetIdentifier, bytes memory _data) public returns (bytes memory) {
         address target = getContractAddress(_targetIdentifier);
         (bool success, bytes memory result) = target.delegatecall(_data);

src/FlashLoan.sol

@@ -19,10 +19,12 @@ import {ITokenSwapsModule} from "./interfaces/ITokenSwapsModule.sol";
 import {IAaveFunctionsModule} from "./interfaces/IAaveFunctionsModule.sol";
 
 // ================================================================
-// │                   AAVEFUNCTIONS CONTRACT                     │
+// │                     FLASH LOAN CONTRACT                      │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Flash Loan for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions for AavePM to execute a flash loan to repay debt and withdraw collateral.
 contract FlashLoan {
     /// @notice Flash loan callback function.
     /// @dev This function is called by the Aave pool contract after the flash loan is executed.

src/FunctionChecks.sol

@@ -12,7 +12,9 @@ import {IAavePM} from "./interfaces/IAavePM.sol";
 // │                   FUNCTION CHECKS CONTRACT                   │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Function Checks for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions used inside AavePM modifiers.
 contract FunctionChecks {
     /// @notice // TODO: Add comment
     function _checkOwner(address _owner) internal view {

src/interfaces/IAaveFunctionsModule.sol

@@ -3,7 +3,8 @@ pragma solidity 0.8.24;
 
 import {IAavePM} from "./IAavePM.sol";
 
-/// @notice // TODO: Add comment
+/// @title AaveFunctionsModule interface
+/// @notice This interface defines the essential structures and functions for the AaveFunctionsModule contract.
 interface IAaveFunctionsModule {
     error AaveFunctionsModule__InvalidAavePMProxyAddress();
 

src/interfaces/IAavePM.sol

@@ -119,8 +119,6 @@ interface IAavePM {
     function deleverage() external;
     function aaveSupplyFromContractBalance() external payable returns (uint256 suppliedCollateral);
     function aaveRepayUSDCFromContractBalance() external;
-
-    /// TODO: Move to different heading
     function delegateCallHelper(string memory _targetIdentifier, bytes memory _data) external returns (bytes memory);
 
     // ================================================================

src/interfaces/IBorrowAndWithdrawUSDCModule.sol

@@ -1,7 +1,8 @@
 // SPDX-License-Identifier: MIT
 pragma solidity 0.8.24;
 
-/// @notice // TODO: Add comment
+/// @title BorrowAndWithdrawUSDCModule interface
+/// @notice This interface defines the essential structures and functions for the BorrowAndWithdrawUSDCModule contract.
 interface IBorrowAndWithdrawUSDCModule {
     error BorrowAndWithdrawUSDCModule__InvalidAavePMProxyAddress();
 

src/interfaces/IRebalanceModule.sol

@@ -1,7 +1,8 @@
 // SPDX-License-Identifier: MIT
 pragma solidity 0.8.24;
 
-/// @notice // TODO: Add comment
+/// @title RebalanceModule interface
+/// @notice This interface defines the essential structures and functions for the RebalanceModule contract.
 interface IRebalanceModule {
     error RebalanceModule__InvalidAavePMProxyAddress();
 

src/interfaces/IReinvestModule.sol

@@ -1,7 +1,8 @@
 // SPDX-License-Identifier: MIT
 pragma solidity 0.8.24;
 
-/// @notice // TODO: Add comment
+/// @title ReinvestModule interface
+/// @notice This interface defines the essential structures and functions for the ReinvestModule contract.
 interface IReinvestModule {
     error ReinvestModule__InvalidAavePMProxyAddress();
 

src/interfaces/ITokenSwapsModule.sol

@@ -4,7 +4,8 @@ pragma solidity 0.8.24;
 import {IAavePM} from "./IAavePM.sol";
 import {ISwapRouter} from "@uniswap/v3-periphery/contracts/interfaces/ISwapRouter.sol";
 
-/// @notice // TODO: Add comment
+/// @title TokenSwapsModule interface
+/// @notice This interface defines the essential structures and functions for the TokenSwapsModule contract.
 interface ITokenSwapsModule {
     error TokenSwapsModule__InvalidAavePMProxyAddress();
 

src/modules/AaveFunctionsModule.sol

@@ -19,10 +19,12 @@ import {ITokenSwapsModule} from "src/interfaces/ITokenSwapsModule.sol";
 import {IAaveFunctionsModule} from "src/interfaces/IAaveFunctionsModule.sol";
 
 // ================================================================
-// │                   AAVEFUNCTIONS CONTRACT                     │
+// │                  AAVE FUNCTIONS MODULE CONTRACT              │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Aave Functions Module for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions for AavePM to interact with the Aave protocol.
 contract AaveFunctionsModule is IAaveFunctionsModule {
     // ================================================================
     // │                         MODULE SETUP                         │

src/modules/BorrowAndWithdrawUSDCModule.sol

@@ -15,10 +15,12 @@ import {IAaveFunctionsModule} from "../interfaces/IAaveFunctionsModule.sol";
 import {IBorrowAndWithdrawUSDCModule} from "../interfaces/IBorrowAndWithdrawUSDCModule.sol";
 
 // ================================================================
-// │               BORROW AND WITHDRAW USDC CONTRACT              │
+// │           BORROW AND WITHDRAW USDC MODULE CONTRACT           │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Borrow and Withdraw USDC Module for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions for AavePM to borrow and withdraw USDC from the Aave protocol.
 contract BorrowAndWithdrawUSDCModule is IBorrowAndWithdrawUSDCModule {
     // ================================================================
     // │                         MODULE SETUP                         │

src/modules/RebalanceModule.sol

@@ -14,10 +14,12 @@ import {IAaveFunctionsModule} from "src/interfaces/IAaveFunctionsModule.sol";
 import {IRebalanceModule} from "src/interfaces/IRebalanceModule.sol";
 
 // ================================================================
-// │                       REBALANCE CONTRACT                     │
+// │                    REBALANCE MODULE CONTRACT                 │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Rebalance Module for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions for AavePM to rebalance the Aave position by repaying debt to increase the health factor.
 contract RebalanceModule is IRebalanceModule {
     // ================================================================
     // │                         MODULE SETUP                         │

src/modules/ReinvestModule.sol

@@ -15,10 +15,13 @@ import {IAaveFunctionsModule} from "../interfaces/IAaveFunctionsModule.sol";
 import {IReinvestModule} from "../interfaces/IReinvestModule.sol";
 
 // ================================================================
-// │                        REINVEST CONTRACT                     │
+// │                     REINVEST MODULE CONTRACT                 │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Aave Functions Module for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions for AavePM to borrow USDC,
+///         swap to wstETH, and supply to Aave maintaining the target health factor.
 contract ReinvestModule is IReinvestModule {
     // ================================================================
     // │                         MODULE SETUP                         │
@@ -35,7 +38,7 @@ contract ReinvestModule is IReinvestModule {
         return VERSION;
     }
 
-    address immutable public aavePMProxyAddress;
+    address public immutable aavePMProxyAddress;
 
     constructor(address _aavePMProxyAddress) {
         aavePMProxyAddress = _aavePMProxyAddress;

src/modules/TokenSwapsModule.sol

@@ -20,10 +20,12 @@ import {IERC20Extended} from "../interfaces/IERC20Extended.sol";
 import {ITokenSwapsModule} from "../interfaces/ITokenSwapsModule.sol";
 
 // ================================================================
-// │                  TokenSwapsModule CONTRACT                   │
+// │                  TOKEN SWAP MODULE CONTRACT                  │
 // ================================================================
 
-/// @notice // TODO: Add comment
+/// @title Token Swap Module for the Aave Position Manager
+/// @author EridianAlpha
+/// @notice This contract contains the functions for AavePM to swap tokens using UniswapV3.
 contract TokenSwapsModule is ITokenSwapsModule {
     // ================================================================
     // │                         MODULE SETUP                         │