diff --git a/database.py b/database.py index f5063f2..046e463 100644 --- a/database.py +++ b/database.py @@ -67,7 +67,7 @@ def get_watchlist(user): if user not in data.keys(): get_stocks(user) - get_watchlist(user) + return get_watchlist(user) return data[user]['watchlist'] @@ -77,7 +77,7 @@ def watch(user, stock, est_price=0): if user not in data.keys(): get_stocks(user) - add_to_watchlist(user, list_name, stock) + return add_to_watchlist(user, list_name, stock) if not stock_exists(stock): return False