diff --git a/.codeocean/app-panel.json b/.codeocean/app-panel.json
index 43b6d07..4feaa51 100644
--- a/.codeocean/app-panel.json
+++ b/.codeocean/app-panel.json
@@ -23,6 +23,24 @@
 			"name": "Basic",
 			"icon": "📂"
 		},
+		{
+			"id": "VisGeneral",
+			"name": "Visualization: General",
+			"description": "General visualization and plotting options",
+			"icon": "\ud83d\udcca"
+		},
+		{
+			"id": "VisBarChart",
+			"name": "Visualization: Bar Chart",
+			"description": "Bar chart visualization options",
+			"icon": "\ud83d\udcca"
+		},
+		{
+			"id": "VisPieChart",
+			"name": "Visualization: Pie Chart",
+			"description": "Pie chart visualization options",
+			"icon": "\ud83d\udcca"
+		},
 		{
 			"id": "FvI4Z2eb9sjL47Jt",
 			"name": "Advanced",
@@ -106,9 +124,13 @@
 			"name": "Round estimates",
 			"param_name": "round_estimates",
 			"description": "Whether to round estimates",
-			"type": "text",
+			"type": "list",
 			"value_type": "string",
-			"default_value": "TRUE"
+			"default_value": "TRUE",
+			"extra_data": [
+				"TRUE",
+				"FALSE"
+			]
 		},
 		{
 			"id": "m9K2pX4bV7nT1eR6",
@@ -160,7 +182,7 @@
 		},
 		{
 			"id": "R8dK2pV5yQ1tS6x9",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisGeneral",
 			"name": "Label font size",
 			"param_name": "label_font_size",
 			"description": "Font size for labels in plots",
@@ -170,7 +192,7 @@
 		},
 		{
 			"id": "p2X7mN4vT6cL1hQ9",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisGeneral",
 			"name": "Label distance",
 			"param_name": "label_distance",
 			"description": "Distance of labels from bars",
@@ -180,7 +202,7 @@
 		},
 		{
 			"id": "h1V6xQ3mT9kB2nR7",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisBarChart",
 			"name": "Y-axis expansion",
 			"param_name": "y_axis_expansion",
 			"description": "Expansion of the y-axis",
@@ -190,7 +212,7 @@
 		},
 		{
 			"id": "K9cP1rV6xT3mQ7nB",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisGeneral",
 			"name": "Fill colors",
 			"param_name": "fill_colors",
 			"description": "Comma-separated fill colors for bars",
@@ -200,17 +222,21 @@
 		},
 		{
 			"id": "L5mT8yQ2vR1nX6cP",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisPieChart",
 			"name": "Pie chart in 3D",
 			"param_name": "pie_chart_in_3d",
 			"description": "Whether to draw pie charts in 3D",
-			"type": "text",
+			"type": "list",
 			"value_type": "string",
-			"default_value": "TRUE"
+			"default_value": "TRUE",
+			"extra_data": [
+				"TRUE",
+				"FALSE"
+			]
 		},
 		{
 			"id": "B6tQ1pV8mR3xN2cL",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisBarChart",
 			"name": "Bar width",
 			"param_name": "bar_width",
 			"description": "Width of bars",
@@ -220,17 +246,21 @@
 		},
 		{
 			"id": "x3M7nQ1tV9kR5cB2",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisBarChart",
 			"name": "Draw bar border",
 			"param_name": "draw_bar_border",
 			"description": "Whether to draw borders around bars",
-			"type": "text",
+			"type": "list",
 			"value_type": "string",
-			"default_value": "TRUE"
+			"default_value": "TRUE",
+			"extra_data": [
+				"TRUE",
+				"FALSE"
+			]
 		},
 		{
 			"id": "Y4nR2pX7mL1vQ6cT",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisGeneral",
 			"name": "Plot type",
 			"param_name": "plot_type",
 			"description": "Plot type: bar or pie",
@@ -240,7 +270,7 @@
 		},
 		{
 			"id": "t6P3mQ9vR1xK7nB2",
-			"category": "FvI4Z2eb9sjL47Jt",
+			"category": "VisGeneral",
 			"name": "Plot titles fontsize",
 			"param_name": "plot_titles_fontsize",
 			"description": "Font size for plot titles",
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 9aa0357..57756db 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Code Ocean capsule - MOSuite - filter differential expression results
 
+## Development version
+
+- Improved the Code Ocean parameter UI for the filter diff capsule (#2, @phoman14).
+
 ## v2.0
 
 - Standardize app panel across MOSuite capsules.
@@ -10,4 +14,4 @@
 
 Initial release
 
-<https://poc-nci.codeocean.io/capsule/0936928/tree/v1> `54a711a`
\ No newline at end of file
+<https://poc-nci.codeocean.io/capsule/0936928/tree/v1> `54a711a`