fixed bug with images in expandable list view

app/src/main/java/com/example/paktalin/lavina/adapters/ExpandableListAdapter.java

@@ -3,6 +3,7 @@ package com.example.paktalin.lavina.adapters;
 import android.content.Context;
 import android.content.Context;
 import android.graphics.Bitmap;
 import android.graphics.Bitmap;
 import android.graphics.Typeface;
 import android.graphics.Typeface;
+import android.util.Log;
 import android.view.LayoutInflater;
 import android.view.LayoutInflater;
 import android.view.View;
 import android.view.View;
 import android.view.ViewGroup;
 import android.view.ViewGroup;
@@ -58,9 +59,17 @@ public class ExpandableListAdapter extends BaseExpandableListAdapter {
         }
         }
 
 
         TextView txtListChild = convertView.findViewById(R.id.lblListItem);
         TextView txtListChild = convertView.findViewById(R.id.lblListItem);
-        if (groupPosition == 0) {
-            ImageView imageView = convertView.findViewById(R.id.image_item);
+
+        String group = (String)getGroup(groupPosition);
+
+        Log.d("TEST", "current group" + group);
+
+        ImageView imageView = convertView.findViewById(R.id.image_item);
+        if (group.equals("COLOR")) {
+            Log.d("TEST", "wohoo");
             imageView.setImageBitmap(colorImages[childPosition]);
             imageView.setImageBitmap(colorImages[childPosition]);
+        } else {
+            imageView.setImageResource(android.R.color.transparent);
         }
         }
 
 
         txtListChild.setText(childText);
         txtListChild.setText(childText);