Initial tracking done, needs check for other twiss parameters in user defined

model.py

@@ -43,6 +43,13 @@ class Model:
     def getInitialEnergy(self):
         return self.om.EnergyAt(self.energyReference)[0]
 
+    def initializeMagnets(self):
+        print('Initializing all magnets to zero')
+        self.om.setRegExpElement('.*', 'MQUA', 'k1', 0)
+        self.om.setRegExpElement('.*', 'MQSK', 'k1', 0)
+        self.om.setRegExpElement('.*', 'MSEX', 'k2', 0)
+
+
     def updateFromMachine(self,machine):
         pol = ['LH', 'LV+', 'LV-', 'C+', 'C-', 'ZL']
         mag = machine['Magnet']