diff --git a/index.html b/index.html
index 311a64d..c408bda 100644
--- a/index.html
+++ b/index.html
@@ -41,7 +41,7 @@ <h2>Load data</h2>
   </p>
 
   <p>
-    <input type="button" class="btn btn-success" value="load data" onclick="document.editor.textbox.value+='\n' + loadData.innerHTML" />
+    <input type="button" class="btn btn-success" value="load data" onclick="document.editor.textbox.value+=loadData" />
   </p>
   
 <hr />
@@ -136,11 +136,11 @@ <h2>Correlation between columns</h2>
 </p>
 
 <p>
-  <input type="button" class="btn btn-primary" value="corr" onclick="document.editor.textbox.value+=document.editor.variable.value === '' ? alert('Please enter a variable name in the \'Load data\' section.') || '' : document.editor.correlation.value === '' ? '\n' + document.editor.variable.value + '.' + this.value + '()' : '\n' + document.editor.correlation.value + ' = ' + document.editor.variable.value + '.' + this.value + '()'" />
-  <input type="button" class="btn btn-primary" value="figure" onclick="document.editor.textbox.value+=document.editor.figure.value === '' ? alert('Please enter a name in the figure field.') || '' : '\n' + document.editor.figure.value + ' = plt.' + this.value + '(figsize=(8,8))'" />
+  <input type="button" class="btn btn-primary" value="corr" onclick="document.editor.textbox.value+=document.editor.variable.value === '' ? alert('Please enter a variable name in the \'variable\' field, in the \'Load data\' section.') || '' : document.editor.correlation.value === '' ? '\n' + document.editor.variable.value + '.' + this.value + '()' : '\n' + document.editor.correlation.value + ' = ' + document.editor.variable.value + '.' + this.value + '()'" />
+  <input type="button" class="btn btn-primary" value="figure" onclick="document.editor.textbox.value+=document.editor.figure.value === '' ? alert('Please enter a name in the \'figure\' field, in the \'Correlation between columns\' section.') || '' : '\n' + document.editor.figure.value + ' = plt.' + this.value + '(figsize=(8,8))'" />
   <input type="button" class="btn btn-primary" value="matshow" onclick="matshow(this.value)" />
-  <input type="button" class="btn btn-primary" value="x ticks" onclick="document.editor.textbox.value+=document.editor.correlation.value === '' ? alert('Please enter a name in the correlation field.') || '' : '\nplt.' + this.value.replace(' ', '') + '(range(len(' + document.editor.correlation.value + '.columns)), ' + document.editor.correlation.value + '.columns, rotation=\'vertical\')'" />
-  <input type="button" class="btn btn-primary" value="y ticks" onclick="document.editor.textbox.value+=document.editor.correlation.value === '' ? alert('Please enter a name in the correlation field.') || '' : '\nplt.' + this.value.replace(' ', '') + '(range(len(' + document.editor.correlation.value + '.columns)), ' + document.editor.correlation.value + '.columns)'" />
+  <input type="button" class="btn btn-primary" value="x ticks" onclick="document.editor.textbox.value+=document.editor.correlation.value === '' ? alert('Please enter a name in the \'figure\' field, in the \'Correlation between columns\' section.') || '' : '\nplt.' + this.value.replace(' ', '') + '(range(len(' + document.editor.correlation.value + '.columns)), ' + document.editor.correlation.value + '.columns, rotation=\'vertical\')'" />
+  <input type="button" class="btn btn-primary" value="y ticks" onclick="document.editor.textbox.value+=document.editor.correlation.value === '' ? alert('Please enter a name in the \'correlation\' field, in the \'Correlation between columns\' section.') || '' : '\nplt.' + this.value.replace(' ', '') + '(range(len(' + document.editor.correlation.value + '.columns)), ' + document.editor.correlation.value + '.columns)'" />
 </p>
 
 <p>
@@ -151,7 +151,7 @@ <h2>Correlation between columns</h2>
 
 <p>
   <input type="button" class="btn btn-primary" value="box" onclick="box(); analysis=this.value" />
-  <input type="button" class="btn btn-primary" value="boxplot cols" onclick="document.editor.textbox.value+=document.editor.boxplotCats.value === '' ? alert('Please enter some comma separated categories.') || '' : '\n' + this.value.replace(' ', '_') + ' = [' + document.editor.boxplotCats.value.split(',').map(el => `'${el.replace(' ', '')}'`).join().replaceAll(',', ', ') + ']'; document.editor.boxplotCats.value=''; analysis=this.value" />
+  <input type="button" class="btn btn-primary" value="boxplot cols" onclick="document.editor.textbox.value+=document.editor.boxplotCats.value === '' ? alert('Please enter some comma separated categories in the \'boxplot categories\' field, in the \'Correlation between columns\' section.') || '' : '\n' + this.value.replace(' ', '_') + ' = [' + document.editor.boxplotCats.value.split(',').map(el => `'${el.replace(' ', '')}'`).join().replaceAll(',', ', ') + ']'; document.editor.boxplotCats.value=''; analysis=this.value" />
   <input class="btn" name="boxplotCats" id="plots" placeholder="boxplot catergories" size="12" type="textfield" />
 </p>
 
@@ -235,28 +235,20 @@ <h3>Same category</h3>
   <input type="button" class="btn btn-column" value="number of records (double)" onclick="numberSelectionDoubleSame();" />
 </p>
 
-  <textarea id="load-data" class="hide">
-conn = sqlite3.connect('data/sakila.db')
-
-df = pd.read_sql('''
-    SELECT
-        rental.rental_id, rental.rental_date, rental.return_date,
-        customer.last_name AS customer_lastname,
-        store.store_id,
-        city.city AS rental_store_city,
-        film.title AS film_title, film.rental_duration AS film_rental_duration,
-        film.rental_rate AS film_rental_rate, film.replacement_cost AS film_replacement_cost,
-        film.rating AS film_rating
-    FROM rental
-    INNER JOIN customer ON rental.customer_id == customer.customer_id
-    INNER JOIN inventory ON rental.inventory_id == inventory.inventory_id
-    INNER JOIN store ON inventory.store_id == store.store_id
-    INNER JOIN address ON store.address_id == address.address_id
-    INNER JOIN city ON address.city_id == city.city_id
-    INNER JOIN film ON inventory.film_id == film.film_id
-    ;
-''', conn, index_col='rental_id', parse_dates=['rental_date', 'return_date'])
-  </textarea>
+<p>
+  <h3>Different category</h3>
+  <input type="button" class="btn btn-column" value="number of records (double)" onclick="numberSelectionDoubleDiff();" />
+  <input type="button" class="btn btn-column" value="number of records (double) is in" onclick="numberSelectionDoubleDiffIsIn();" />
+</p>
+
+<p>
+  <input class="btn" name="groupA" id="groupA" placeholder="group A" size="10" type="textfield" />
+  <select id="selectionGrouping" name="selectionGrouping">
+    <option value="&">and</option>
+    <option value="|">or</option>
+  </select>
+  <input class="btn" name="groupB" id="groupB" placeholder="group B" size="10" type="textfield" />
+</p>
 
 </div>
           </td>
diff --git a/script.js b/script.js
index 6781f15..2c73e69 100644
--- a/script.js
+++ b/script.js
@@ -1,11 +1,32 @@
 
 // Load data
 let variable = document.getElementById("variable");
-let loadData = document.getElementById("load-data");
+let loadData = `
+conn = sqlite3.connect('data/sakila.db')
+
+df = pd.read_sql('''
+    SELECT
+        rental.rental_id, rental.rental_date, rental.return_date,
+        customer.last_name AS customer_lastname,
+        store.store_id,
+        city.city AS rental_store_city,
+        film.title AS film_title, film.rental_duration AS film_rental_duration,
+        film.rental_rate AS film_rental_rate, film.replacement_cost AS film_replacement_cost,
+        film.rating AS film_rating
+    FROM rental
+    INNER JOIN customer ON rental.customer_id == customer.customer_id
+    INNER JOIN inventory ON rental.inventory_id == inventory.inventory_id
+    INNER JOIN store ON inventory.store_id == store.store_id
+    INNER JOIN address ON store.address_id == address.address_id
+    INNER JOIN city ON address.city_id == city.city_id
+    INNER JOIN film ON inventory.film_id == film.film_id
+    ;
+''', conn, index_col='rental_id', parse_dates=['rental_date', 'return_date'])
+`;
 
 const data = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else {
       document.editor.textbox.value+= "\n" + document.editor.variable.value + "." + stat;
   }
@@ -19,7 +40,7 @@ let mostValueCounts = document.getElementById("mostValueCounts");
 
 const describe = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else {
       document.editor.textbox.value+= column.value === "" ? "\n" + document.editor.variable.value + "." + stat + "()\n" : "\n" + document.editor.variable.value + "['" + document.editor.column.value + "']." + stat + "()\n";
   }
@@ -27,9 +48,9 @@ const describe = (stat) => {
 
 const meanMedian = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (column.value === "") {
-      return alert("Please add a column name in the 'Numerical analysis and visualisation' section.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else {
       document.editor.textbox.value+= "\n" + variable.value + "['" + column.value + "']." + stat + "()";
   }
@@ -37,9 +58,9 @@ const meanMedian = (stat) => {
 
 const analyseBox = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (column.value === "") {
-      return alert("Please add a column name in the 'Numerical analysis and visualisation' section.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else if (axis.value !== "") {
       document.editor.textbox.value+= "\n" + axis.value + " = " + variable.value + "['" + column.value + "'].plot(kind='" + stat + "', vert=False, figsize=(14,6))";
   } else {
@@ -49,9 +70,9 @@ const analyseBox = (stat) => {
 
 const densityHist = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (column.value === "") {
-      return alert("Please add a column name in the 'Numerical analysis and visualisation' section.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else if (axis.value !== "") {
       document.editor.textbox.value+= "\n" + axis.value + " = " + variable.value + "['" + column.value + "'].plot(kind='" + stat + "', figsize=(14,6))";
   } else {
@@ -61,21 +82,21 @@ const densityHist = (stat) => {
 
 const colour = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (analysis === undefined) {
       return alert("Please select an analysis type.");
   } else if (column.value === "") {
-      return alert("Please add a column name in the 'Numerical analysis and visualisation' section.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else if (axis.value === "") {
-      return alert("Please add an axis name in the 'Numerical analysis and visualisation' section.");
+      return alert("Please enter an axis name in the 'axis' field, in the 'Numerical analysis and visualisation' section.");
   } else {
-      document.editor.textbox.value+='\n' + document.editor.axis.value + '.axvline(' + document.editor.variable.value + '[\'' + document.editor.column.value + '\'].' + analysis + '(), color=\'' + document.editor.colours.value + '\')'
+      document.editor.textbox.value+="\n" + document.editor.axis.value + ".axvline(" + document.editor.variable.value + "['" + document.editor.column.value + "']." + analysis + "(), color='" + document.editor.colours.value + "')";
   }
 }
 
 const round = () => {
   if (mostValueCounts.value === "") {
-    return alert("Please add a number in the 'Categorical analysis and visualisation' section.");
+    return alert("Please add a number in the 'number' field, in the 'Categorical analysis and visualisation' section.");
   } else {
     
       document.editor.textbox.value+='.round(' + mostValueCounts.value + ')';
@@ -84,9 +105,9 @@ const round = () => {
 
 const axisLabel = (stat) => {
   if (axis.value === "") {
-      return alert("Please enter an axis name in the axis field.");
+      return alert("Please enter an axis name in the 'axis' field, in the 'Numerical analysis and visualisation' section.");
   } else if (label.value === '') {
-    return alert("Please enter a label name in the label field to the right.");
+    return alert("Please enter a label name in the 'label' field, in the 'Numerical analysis and visualisation' section.");
   } else {
       document.editor.textbox.value+= "\n" + document.editor.axis.value + ".set_" + stat.replace(" ", "") + "('" + document.editor.label.value + "')";
   }
@@ -95,45 +116,45 @@ const axisLabel = (stat) => {
 // Categorical analysis and visualisation
 const categoricalValueCounts = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (column.value === "") {
-      return alert("Please add a column name.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else {
      let head = mostValueCounts.value;
      if (head) {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + column.value + '\'].' + stat.replace(' ', '_') + '().head(' + head + ')';
+      document.editor.textbox.value+="\n" + variable.value + "['" + column.value + "']." + stat.replace(' ', '_') + "().head(" + head + ")";
      } else {
-        document.editor.textbox.value+='\n' + variable.value + '[\'' + column.value + '\'].' + stat.replace(' ', '_') + '()'
+        document.editor.textbox.value+="\n" + variable.value + "['" + column.value + "']." + stat.replace(' ', '_') + "()";
      }
   }
 }
 
 const categoricalBar = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (column.value === "") {
-      return alert("Please add a column name.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else {
      let head = mostValueCounts.value;
      if (head) {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + column.value + '\'].value_counts().head(' + head + ').plot(kind=\'' + stat + '\', figsize=(14,6))';
+      document.editor.textbox.value+="\n" + variable.value + "['" + column.value + "'].value_counts().head(" + head + ").plot(kind='" + stat + "', figsize=(14,6))";
      } else {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + column.value + '\'].value_counts().plot(kind=\'' + stat + '\', figsize=(14,6))';
+      document.editor.textbox.value+="\n" + variable.value + "['" + column.value + "'].value_counts().plot(kind='" + stat + "', figsize=(14,6))";
      }
   }
 }
 
 const categoricalPie = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (column.value === "") {
-      return alert("Please add a column name.");
+      return alert("Please add a column name in the 'column' field, in the 'Numerical analysis and visualisation' section.");
   } else {
      let head = mostValueCounts.value;
      if (head) {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + column.value + '\'].value_counts().head(' + head + ').plot(kind=\'' + stat + '\', figsize=(6,6))';
+      document.editor.textbox.value+="\n" + variable.value + "['" + column.value + "'].value_counts().head(" + head + ").plot(kind='" + stat + "', figsize=(6,6))";
      } else {
-        document.editor.textbox.value+='\n' + variable.value + '[\'' + column.value + '\'].value_counts().plot(kind=\'' + stat + '\', figsize=(6,6))';
+        document.editor.textbox.value+="\n" + variable.value + "['" + column.value + "'].value_counts().plot(kind='" + stat + "', figsize=(6,6))";
      }
   }
 }
@@ -158,51 +179,51 @@ let groupB = document.getElementById("groupB");
 
 const matshow = (stat) => {
   if (correlation.value === "") {
-    return alert('Please enter a name in the correlation field.');
+    return alert("Please enter a name in the 'correlation field', in the 'Correlation between columns' section.");
   } else if (figure.value === "") {
-      return alert('Please enter a name in the figure field.');
+      return alert("Please enter a name in the 'figure' field, in the 'Correlation between columns' section.");
   } else {
-      document.editor.textbox.value+='\nplt.' + stat + '(' + document.editor.correlation.value + ', cmap=\'RdBu\', fignum=' + document.editor.figure.value + '.number)'
+      document.editor.textbox.value+="\nplt." + stat + "(" + document.editor.correlation.value + ", cmap='RdBu', fignum=" + document.editor.figure.value + ".number)";
   }
 }
 
 const scatters = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (scatterX.value === "") {
-      return alert("Please enter a value in the 'scatter x' field.");
+      return alert("Please enter a value in the 'scatter x' field, in the 'Correlation between columns' section.");
   } else if (scatterY.value === "") {
-      return alert("Please enter a value in the 'scatter y' field.");
+      return alert("Please enter a value in the 'scatter y' field, in the 'Correlation between columns' section.");
   } else {
-      return document.editor.textbox.value+='\n' + document.editor.variable.value + '.plot(kind=\'' + stat + '\', x=\'' + document.editor.scatterX.value + '\', y=\'' + document.editor.scatterY.value + '\', figsize=(6,6))';
+      return document.editor.textbox.value+="\n" + document.editor.variable.value + ".plot(kind='" + stat + "', x='" + document.editor.scatterX.value + "', y='" + document.editor.scatterY.value + "', figsize=(6,6))";
   }
 }
 
 const boxPlot = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (plots.value === "") {
-      return alert("Please enter at least one category into the 'boxplot categories' field.");
+      return alert("Please enter at least one category into the 'boxplot categories' field, in the 'Correlation between columns' section.");
   } else if (boxPlotAxis.value === "") {
-      return alert("Please enter a name in the 'boxplot axis' field.");
+      return alert("Please enter a name in the 'boxplot axis' field, in the 'Correlation between columns' section.");
   } else {
-      return document.editor.textbox.value+= '\n' + boxPlotAxis.value + ' = ' + document.editor.variable.value + '[[' + document.editor.boxplotCats.value.split(',').map(el => `'${el.replace(' ', '')}'`).join().replaceAll(',', ', ') + ']].boxplot(by=\'' + document.editor.boxplotCats.value.split(',')[document.editor.boxplotCats.value.split(',').length - 1].replace(' ', '') + '\', figsize=(14,6))\n' + boxPlotAxis.value + '.set_ylabel(\'' + document.editor.boxplotCats.value.split(',')[0] + '\')';
+      return document.editor.textbox.value+="\n" + boxPlotAxis.value + " = " + document.editor.variable.value + "[[" + document.editor.boxplotCats.value.split(',').map(el => `'${el.replace(' ', '')}'`).join().replaceAll(',', ', ') + "]].boxplot(by='" + document.editor.boxplotCats.value.split(',')[document.editor.boxplotCats.value.split(',').length - 1].replace(' ', '') + "', figsize=(14,6))\n" + boxPlotAxis.value + ".set_ylabel('" + document.editor.boxplotCats.value.split(',')[0] + "')";
   }
 }
 
 const boxPlotGrouped = () => {
   if (variable.value === '') {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (boxPlotCategory.value === '') {
-      return alert("Please enter a category in the 'category' field to the right.");
+      return alert("Please enter a category in the 'category' field, in the 'Correlation between columns' section.");
   } else if (boxPlotCategoryValue.value === '') {
-      return alert("Please enter a category value in the 'value' field to the right.");
+      return alert("Please enter a category value in the 'value' field, in the 'Correlation between columns' section.");
   } else if (boxPlotCategoryColumnA.value === '') {
-      return alert("Please enter a category in the 'column A' field to the right.");
+      return alert("Please enter a category in the 'column A' field, in the 'Correlation between columns' section.");
   } else if (boxPlotCategoryColumnB.value === '') {
-      return alert("Please enter a category in the 'column B' field to the right.");
+      return alert("Please enter a category in the 'column B' field, in the 'Correlation between columns' section.");
   } else if (boxPlotAxis.value === '') {
-      return alert("Please enter a name in the 'boxplot axis' field.");
+      return alert("Please enter a name in the 'boxplot axis' field, in the 'Correlation between columns' section.");
   } else {
       return document.editor.textbox.value+= "\n" + boxPlotAxis.value + " = " + variable.value + ".loc[" + variable.value + "['" + boxPlotCategory.value + "'] == " + boxPlotCategoryValue.value + ", ['" + boxPlotCategoryColumnA.value + "', '" + boxPlotCategoryColumnB.value + "']]" + "\n" + boxPlotAxis.value + ".boxplot(by='" + boxPlotCategoryColumnB.value + "', figsize=(14,6))";
   }
@@ -210,11 +231,11 @@ const boxPlotGrouped = () => {
 
 const box = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section.");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (plots.value === "") {
-      return alert("Please enter at least one category into the 'boxplot categories' field.");
+      return alert("Please enter at least one category into the 'boxplot categories' field, in the 'Correlation between columns' section.");
   } else {
-      return document.editor.textbox.value+= '\n' + document.editor.variable.value + '[[' + document.editor.boxplotCats.value.split(',').map(el => `'${el.replace(' ', '')}'`) + ']].plot(kind=\'box\', subplots=True, layout=(2,3), figsize=(14,8))';
+      return document.editor.textbox.value+="\n" + document.editor.variable.value + "[[" + document.editor.boxplotCats.value.split(',').map(el => `'${el.replace(' ', '')}'`) + "]].plot(kind='box', subplots=True, layout=(2,3), figsize=(14,8))";
   }
 }
 
@@ -229,37 +250,37 @@ const columnWrangle = (stat) => {
   }
 
   if (variable.value === '') {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (newColumn.value === "") {
-      return alert("Please add a new column name.");
+      return alert("Please add a column name in the 'new column' field in the 'Column Wrangling' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (columnB.value === "") {
-      return alert("Please add a secondary column name in the 'Column Wrangling' section.");
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
   } else {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + newColumn.value + '\'] = ' + variable.value + '[\'' + columnA.value + '\'] ' + stat + ' ' + variable.value + '[\'' + columnB.value + '\']\n' + variable.value + '[\'' + newColumn.value + '\'].head()' + '\n' + variable.value + '[\'' + newColumn.value + '\'].plot(kind=\'hist\',' + bin + ' figsize=(14,6))';
+      document.editor.textbox.value+="\n" + variable.value + "['" + newColumn.value + "'] = " + variable.value + "['" + columnA.value + "'] " + stat + " " + variable.value + "['" + columnB.value + "']\n" + variable.value + "['" + newColumn.value + "'].head()" + "\n" + variable.value + "['" + newColumn.value + "'].plot(kind='hist'," + bin + " figsize=(14,6))";
   }
 }
 
 const columnPlot = (stat) => {
   if (variable.value === "") {
-      return alert("Please enter a variable name in the 'Load data' section");
+      return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (newColumn.value === "") {
-      return alert("Please add a new column name.");
+      return alert("Please add a column name in the 'new column' field in the 'Column Wrangling' section.");
   } else {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + newColumn.value + '\'].plot(kind=\'' + stat + '\', figsize=(14,6))';
+      document.editor.textbox.value+="\n" + variable.value + "['" + newColumn.value + "'].plot(kind='" + stat + "', figsize=(14,6))";
   }
 }
 
 const columnScatter = (stat) => {
   if (variable.value === '') {
-    return alert('Please enter a variable name in the \'Load data\' section');
+    return alert("Please enter a variable name in the 'Load data' section");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (scatterAxis.value === "") {
-      return alert("Please add a scatter column name.");
+      return alert("Please add a scatter column name in the 'scatter' field, in the 'Add and calculate a new column' section.");
   } else {
-      document.editor.textbox.value+='\n' + variable.value + '.plot(kind=\'' + stat + '\', x=\'' + columnA.value + '\', y=\'' + scatterAxis.value + '\', figsize=(6,6))';
+      document.editor.textbox.value+="\n" + variable.value + ".plot(kind='" + stat + "', x='" + columnA.value + "', y='" + scatterAxis.value + "', figsize=(6,6))";
   }
 }
 
@@ -268,26 +289,26 @@ let modify = document.getElementById("modify");
 
 const columnWrangleModify = (stat) => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (modify.value === "") {
-      return alert("Please add a value in the modify field to the right.");
+      return alert("Please add a value in the 'modify' field, in the 'Modify existing column' section.");
   } else if (!Number(modify.value) && Number(modify.value) != "0") {
-      return alert("Please add a valid number in the modify field to the right.");
+      return alert("Please add a valid number in the 'modify' field, in the 'Modify existing column'.");
   } else {
-      document.editor.textbox.value+='\n' + variable.value + '[\'' + columnA.value + '\'] ' + stat + '= ' + modify.value + '\n' + variable.value + '[\'' + columnA.value + '\'].head()';
+      document.editor.textbox.value+="\n" + variable.value + "['" + columnA.value + "'] " + stat + "= " + modify.value + "\n" + variable.value + "['" + columnA.value + "'].head()";
   }
 }
 
 // Selection and indexing
 const individualSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (columnB.value === "") {
-      return alert("Please add a secondary column name in the 'Column Wrangling' section.");
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[" + variable.value + "['" + columnA.value + "'] == '" + columnB.value + "']" + "\n" + variable.value + "['" + columnA.value + "'].head()";
   }
@@ -295,31 +316,31 @@ const individualSelection = () => {
 
 const groupSelectionMean = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (groupA.value === "") {
-      return alert("Please add a category in group A.");
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
   } else if (groupB.value === "") {
-      return alert("Please add a category in group B.");
+      return alert("Please add a category in 'group B' field, in the 'Different category' subsection.");
   } else {
-      document.editor.textbox.value+='\n' + variable.value + '.loc[' + variable.value + '[\'' + columnA.value + '\'] == \'' + groupA.value + '\', \'' + groupB.value + '\'].mean()';
+      document.editor.textbox.value+="\n" + variable.value + ".loc[" + variable.value + "['" + columnA.value + "'] == '" + groupA.value + "', '" + groupB.value + "'].mean()";
   }
 }
 
 const meanSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (groupA.value === "") {
-      return alert("Please add a category in group A.");
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
   } else if (columnB.value === "") {
-      return alert("Please add a secondary column name in the 'Column Wrangling' section.");
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
   } else if (groupB.value === "") {
-      return alert("Please add a category in group B.");
+      return alert("Please add a category in 'group B' field, in the 'Different category' subsection.");
   } else if (newColumn.value === "") {
-      return alert("Please add a new column name.");
+      return alert("Please add a column name in the 'new column' field in the 'Column Wrangling' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[(" + variable.value + "['" + columnA.value + "'] == '" + groupA.value + "') & (" + variable.value + "['" + columnB.value + "'] == '" + groupB.value + "'), '" + newColumn.value + "'].mean()";
   }
@@ -327,13 +348,13 @@ const meanSelection = () => {
 
 const meanMaxSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (mostValueCounts.value === "") {
-      return alert("Please add a number in the 'Categorical analysis and visualisation' section.");
+      return alert("Please add a number in the 'number' field, in the 'Categorical analysis and visualisation' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (columnB.value === "") {
-      return alert("Please add a secondary column name in the 'Column Wrangling' section.");
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[" + variable.value + "['"   + columnA.value + "'] > " + mostValueCounts.value + ", '"   + columnB.value + "'].mean()";
   }
@@ -341,11 +362,11 @@ const meanMaxSelection = () => {
 
 const meanMinSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (mostValueCounts.value === "") {
-      return alert("Please add a number in the 'Categorical analysis and visualisation' section.");
+      return alert("Please add a number in the 'number' field, in the 'Categorical analysis and visualisation' section.");
   } else if (columnB.value === "") {
-      return alert("Please add a secondary column name in the 'Column Wrangling' section.");
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[" + variable.value + "['"   + columnA.value + "'] < " + mostValueCounts.value + ", '"   + columnB.value + "'].mean()";
   }
@@ -353,15 +374,15 @@ const meanMinSelection = () => {
 
 const percentageSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (groupA.value === "") {
-      return alert("Please add a category in group A.");
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
   } else if (newColumn.value === "") {
-      return alert("Please add a new column name.");
+      return alert("Please add a column name in the 'new column' field in the 'Column Wrangling' section.");
   } else if (selectionPercentage.value === "") {
-      return alert("Please select a percentage.");
+      return alert("Please add a percentage in the 'percentage' field, in the 'Mean' subsection.");
   } else {
       let percentage = selectionPercentage.value;
       let percentageNegative = false;
@@ -392,7 +413,7 @@ const percentageSelection = () => {
 
 const maxSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[" + variable.value + "['"   + columnA.value + "'] == " + variable.value + "['"   + columnA.value + "'].max()]";
   }
@@ -400,9 +421,9 @@ const maxSelection = () => {
 
 const sort = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (mostValueCounts.value === "") {
-      return alert("Please add a number in the 'Categorical analysis and visualisation' section.");
+      return alert("Please add a number in the 'number' field, in the 'Categorical analysis and visualisation' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".sort_values(['" + columnA.value + "'], ascending=False).head(" + mostValueCounts.value + ")";
   }
@@ -410,15 +431,13 @@ const sort = () => {
 
 const valueCountSelection = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (groupA.value === "") {
-      return alert("Please add a category in group A.");
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
   } else if (newColumn.value === "") {
-      return alert("Please add a new column name.");
-  } else if (selectionPercentage.value === "") {
-      return alert("Please select a percentage.");
+      return alert("Please add a column name in the 'new column' field in the 'Column Wrangling' section.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[" + variable.value + "['" + columnA.value + "'] == '" + groupA.value + "', '" + newColumn.value + "'].value_counts()";
   }
@@ -426,11 +445,11 @@ const valueCountSelection = () => {
 
 const numberSelectionSingle = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (groupA.value === "") {
-      return alert("Please add a category in group A.");
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[(" + variable.value + "['" + columnA.value + "'] == '" + groupA.value + "')].shape[0]";
   }
@@ -438,14 +457,46 @@ const numberSelectionSingle = () => {
 
 const numberSelectionDoubleSame = () => {
   if (variable.value === "") {
-    return alert("Please enter a variable name in the 'Load data' section");
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
   } else if (columnA.value === "") {
-      return alert("Please add a primary column name in the 'Column Wrangling' section.");
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
   } else if (groupA.value === "") {
-      return alert("Please add a category in group A.");
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
   } else if (groupB.value === "") {
-      return alert("Please add a category in group B.");
+      return alert("Please add a category in 'group B' field, in the 'Different category' subsection.");
   } else {
       document.editor.textbox.value+="\n" + variable.value + ".loc[(" + variable.value + "[" + columnA.value + "'] == '" + groupA.value + "') " + selectionGrouping.value +  " (" + variable.value + "['" + columnA.value + "'] == '" + groupB.value + "')].shape[0]";
   }
 }
+
+const numberSelectionDoubleDiff = () => {
+  if (variable.value === "") {
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
+  } else if (columnA.value === "") {
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
+  } else if (columnB.value === "") {
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
+  } else if (groupA.value === "") {
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
+  } else if (groupB.value === "") {
+      return alert("Please add a category in 'group B' field, in the 'Different category' subsection.");
+  } else {
+      document.editor.textbox.value+="\n" + variable.value + ".loc[(" + variable.value + "['" + columnA.value + "'] == '" + groupA.value + "') " + selectionGrouping.value + " (" + variable.value + "['" + columnB.value + "'] == '" + groupB.value + "')].shape[0]";
+  }
+}
+
+const numberSelectionDoubleDiffIsIn = () => {
+  if (variable.value === "") {
+    return alert("Please enter a variable name in the 'variable' field, in the 'Load data' section.");
+  } else if (columnA.value === "") {
+      return alert("Please add a primary column name in the 'primary column' field, in the 'Column Wrangling' section.");
+  } else if (columnB.value === "") {
+      return alert("Please add a secondary column name in the 'secondary column' field, in the 'Column Wrangling' section.");
+  } else if (groupA.value === "") {
+      return alert("Please add a category in 'group A' field, in the 'Different category' subsection.");
+  } else if (groupB.value === "") {
+      return alert("Please add a category in 'group B' field, in the 'Different category' subsection.");
+  } else {
+      document.editor.textbox.value+="\n" + variable.value + ".loc[(" + variable.value + "['" + columnA.value + "'] == '" + groupA.value + "') " + selectionGrouping.value + " (" + variable.value + "['" + columnB.value + "'].isin([" + groupB.value.split(',').map(el => `'${el.replace(' ', '')}'`).join().replaceAll(',', ', ') + "]" + "))].shape[0]";
+  }
+}