diff --git a/stock.py b/stock.py index 134d200..1fb01ca 100644 --- a/stock.py +++ b/stock.py @@ -1,4 +1,3 @@ - class Stock: def __init__(self, name, shares, price): self.name = name @@ -7,3 +6,8 @@ class Stock: def cost(self): return self.shares * self.price + + def sell(self, num): + self.shares -= num + if self.shares < 0: + self.shares = 0