From 0bafdcb1f7a1b2971aeae14e1294893f04cacbea Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Einar=20Sk=C3=B8ien?= <einaskoi@stud.ntnu.no>
Date: Mon, 23 Feb 2026 10:44:36 +0100
Subject: [PATCH] implement logic in Portfolio class

---
 .../idi/idatt2003/gruppe42/Model/Portfolio.java    | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Portfolio.java b/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Portfolio.java
index fcb26a0..0c39fb4 100644
--- a/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Portfolio.java
+++ b/src/main/java/edu/ntnu/idi/idatt2003/gruppe42/Model/Portfolio.java
@@ -23,7 +23,11 @@ public Portfolio() {
    * @return true or false based on if the operation was successful or not.
    */
   public boolean addShare(Share share) {
-    return true;
+    if (share == null) {
+      return false;
+    } else {
+      return shares.add(share);
+    }
   }
 
   /**
@@ -32,7 +36,11 @@ public boolean addShare(Share share) {
    * @return true or false based on if the operation was successful or not.
    */
   public boolean removeShare(Share share) {
-    return false;
+    if (share == null || !shares.contains(share)) {
+      return false;
+    } else {
+      return shares.remove(share);
+    }
   }
 
   public List<Share> getShares() {
@@ -45,6 +53,6 @@ public List<Share> getShares() {
    * @return true or false based on if the share was found or not.
    */
   public boolean contains(Share share) {
-    return true;
+    return share != null && shares.contains(share);
   }
 }