1 files changed, 14 insertions, 1 deletions
diff --git a/grapher/Models/Options/AccelOptions.cs b/grapher/Models/Options/AccelOptions.cs
index b233552..cd7c4e5 100644
--- a/grapher/Models/Options/AccelOptions.cs
+++ b/grapher/Models/Options/AccelOptions.cs
@@ -1,4 +1,5 @@
 using grapher.Layouts;
+using grapher.Models.Options;
 using System;
 using System.Collections.Generic;
 using System.Linq;
@@ -22,6 +23,8 @@ namespace grapher
             new LogLayout(),
             new SigmoidLayout(),
             new PowerLayout(),
+            new NaturalGainLayout(),
+            new SigmoidGainLayout(),
             new OffLayout()
         }.ToDictionary(k => k.Name);
 
@@ -29,7 +32,8 @@ namespace grapher
             ComboBox accelDropdown,
             Option[] options,
             OptionXY[] optionsXY,
-            Button writeButton)
+            Button writeButton,
+            ActiveValueLabel activeValueLabel)
         {
             AccelDropdown = accelDropdown;
             AccelDropdown.Items.Clear();
@@ -49,6 +53,7 @@ namespace grapher
             Options = options;
             OptionsXY = optionsXY;
             WriteButton = writeButton;
+            ActiveValueLabel = activeValueLabel;
 
             Layout("Default");
         }
@@ -59,10 +64,18 @@ namespace grapher
 
         public int AccelerationIndex { get; private set; }
 
+        public ActiveValueLabel ActiveValueLabel { get; }
+
         public Option[] Options { get; }
 
         public OptionXY[] OptionsXY { get; }
 
+        public void SetActiveValue(int index)
+        {
+            var name = AccelerationTypes.Where(t => t.Value.Index == index).FirstOrDefault().Value.Name;
+            ActiveValueLabel.SetValue(name);
+        }
+
         private void OnIndexChanged(object sender, EventArgs e)
         {
             var accelerationTypeString = AccelDropdown.SelectedItem.ToString();