diff --git a/src/streamlit_app.py b/src/streamlit_app.py
index 209a688..cfe59f5 100644
--- a/src/streamlit_app.py
+++ b/src/streamlit_app.py
@@ -2,6 +2,7 @@ import streamlit as st
 import pandas as pd
 from datetime import datetime
 import pytz
+from db.db_connection import create_client
 from trading.journal import (
     create_trades_table, get_open_trades, get_trade_history,
     add_trade, update_trade, delete_trade, TradeEntry,
@@ -12,7 +13,8 @@ from trading.trading_plan import (
     delete_trading_plan,
     TradingPlan, PlanStatus, Timeframe, MarketFocus, TradeFrequency,
     create_trading_plan_table, save_trading_plan, get_trading_plan,
-    get_all_trading_plans, update_trading_plan
+    get_all_trading_plans, update_trading_plan, get_plan_trades,
+    link_trades_to_plan, calculate_plan_metrics
 )
 from trading.position_calculator import PositionCalculator
 from screener.scanner_controller import run_technical_scanner