diff --git a/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/Picture.java b/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/Picture.java
index 9470184..2f1ef99 100644
--- a/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/Picture.java
+++ b/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/Picture.java
@@ -28,15 +28,10 @@ import lombok.Getter;
 import javax.imageio.ImageIO;
 import java.awt.*;
 import java.awt.image.BufferedImage;
-import java.io.ByteArrayInputStream;
-import java.io.ByteArrayOutputStream;
-import java.io.IOException;
+import java.io.*;
 import java.net.URL;
-import java.util.ArrayList;
-import java.util.List;
 import java.util.concurrent.CompletableFuture;
 
-
 public class Picture {
 
     @Getter
@@ -49,7 +44,6 @@ public class Picture {
     }
 
     public static Picture map(io.github.jwdeveloper.tiktok.messages.data.Image profilePicture) {
-
         var index = profilePicture.getUrlListCount() - 1;
         if (index < 0) {
             return new Picture("");
@@ -74,12 +68,11 @@ public class Picture {
         return CompletableFuture.supplyAsync(this::downloadImage);
     }
 
-    private BufferedImage download(String urlString)
-    {
-        if(urlString.isEmpty())
-        {
+    private BufferedImage download(String urlString) {
+        if (urlString.isEmpty()) {
             return null;
         }
+
         var baos = new ByteArrayOutputStream();
         try (var is = new URL(urlString).openStream()) {
             var byteChunk = new byte[4096];
@@ -103,4 +96,9 @@ public class Picture {
     public static Picture Empty() {
         return new Picture("");
     }
-}
+
+    @Override
+    public String toString() {
+        return "Picture{link='" + link + "', image=" + image + "}";
+    }
+}
\ No newline at end of file
diff --git a/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/badges/Badge.java b/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/badges/Badge.java
index ac429b8..8096934 100644
--- a/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/badges/Badge.java
+++ b/API/src/main/java/io/github/jwdeveloper/tiktok/data/models/badges/Badge.java
@@ -37,6 +37,4 @@ public class Badge {
     public static Badge empty() {
         return new Badge();
     }
-
-
-}
+}
\ No newline at end of file
diff --git a/Client/src/main/java/io/github/jwdeveloper/tiktok/gifts/TikTokGiftManager.java b/Client/src/main/java/io/github/jwdeveloper/tiktok/gifts/TikTokGiftManager.java
index 20a6e19..ccb442f 100644
--- a/Client/src/main/java/io/github/jwdeveloper/tiktok/gifts/TikTokGiftManager.java
+++ b/Client/src/main/java/io/github/jwdeveloper/tiktok/gifts/TikTokGiftManager.java
@@ -22,14 +22,12 @@
  */
 package io.github.jwdeveloper.tiktok.gifts;
 
-import io.github.jwdeveloper.tiktok.data.models.gifts.Gift;
 import io.github.jwdeveloper.tiktok.data.models.Picture;
+import io.github.jwdeveloper.tiktok.data.models.gifts.Gift;
 import io.github.jwdeveloper.tiktok.exceptions.TikTokLiveException;
 import io.github.jwdeveloper.tiktok.live.GiftManager;
 import sun.misc.Unsafe;
 
-import java.lang.reflect.Field;
-import java.lang.reflect.Modifier;
 import java.util.*;
 import java.util.logging.Logger;
 
@@ -89,23 +87,17 @@ public class TikTokGiftManager implements GiftManager {
     }
 
     public Gift findById(int giftId) {
-        if (!indexById.containsKey(giftId)) {
-            return Gift.UNDEFINED;
-        }
-        return indexById.get(giftId);
+        Gift gift = indexById.get(giftId);
+        return gift == null ? Gift.UNDEFINED : gift;
     }
 
     public Gift findByName(String giftName) {
-        if (!indexByName.containsKey(giftName)) {
-            return Gift.UNDEFINED;
-        }
-        return indexByName.get(giftName);
+        Gift gift = indexByName.get(giftName);
+        return gift == null ? Gift.UNDEFINED : gift;
     }
 
     @Override
-    public List<Gift> getGifts()
-    {
+    public List<Gift> getGifts() {
         return indexById.values().stream().toList();
     }
-
-}
+}
\ No newline at end of file