diff --git a/src/onchain/TestArbitrage.sol b/src/onchain/TestArbitrage.sol
index 88ad1fe..42f2f2f 100644
--- a/src/onchain/TestArbitrage.sol
+++ b/src/onchain/TestArbitrage.sol
@@ -115,6 +115,18 @@ contract TestArbitrage is IFlashLoanRecipient, ReentrancyGuard, Ownable, Pausabl
         uint256 lastTradeTimestamp;
     }
 
+    //////////////////////////////////////////////////////////////
+    //                        STATE VARIABLES                 //
+    //////////////////////////////////////////////////////////////
+
+    /// @notice Current contract statistics
+    /// @dev Updated after each successful trade
+    ContractStats public stats;
+
+    /// @notice Mapping to track trade history by ID
+    /// @dev Allows retrieval of specific trade details
+    mapping(uint256 => TradeResult) public tradeHistory;
+
     //////////////////////////////////////////////////////////////
     //                        CONSTRUCTOR                     //
     //////////////////////////////////////////////////////////////