diff --git a/src/onchain/TestArbitrage.sol b/src/onchain/TestArbitrage.sol
index 42f2f2f..e21fa27 100644
--- a/src/onchain/TestArbitrage.sol
+++ b/src/onchain/TestArbitrage.sol
@@ -127,6 +127,14 @@ contract TestArbitrage is IFlashLoanRecipient, ReentrancyGuard, Ownable, Pausabl
     /// @dev Allows retrieval of specific trade details
     mapping(uint256 => TradeResult) public tradeHistory;
 
+    /// @notice Counter for trade IDs
+    /// @dev Incremented for each new trade
+    uint256 public tradeCounter;
+
+    /// @notice Test mode flag for simulation purposes
+    /// @dev When true, enables additional testing features
+    bool public testMode;
+
     //////////////////////////////////////////////////////////////
     //                        CONSTRUCTOR                     //
     //////////////////////////////////////////////////////////////