diff --git a/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py b/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py
index 0e35eacf..65e47f77 100644
--- a/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py
+++ b/simpegPF/Dev/MAG/Intgrl_MAG_Inv_Driver.py
@@ -4,7 +4,7 @@ import pylab as plt
 
 import os
 
-driver = PF.MagneticsIO.MagneticsDriver_Inv('PYMAG3D_inv.inp')
+driver = PF.MagneticsDriver.MagneticsDriver_Inv('PYMAG3D_inv.inp')
 mesh = driver.mesh
 survey = driver.survey
 
@@ -27,12 +27,6 @@ actvMap = Maps.InjectActiveCells(mesh, actv, -100)
 # Creat reduced identity map
 idenMap = Maps.IdentityMap(nP = nC)
 
-# Load starting model file
-mstart = driver.m0
-
-# Load reference file
-mref = driver.mref
-
 # Get magnetization vector for MOF
 M_xyz = driver.magnetizationModel
 
@@ -47,7 +41,7 @@ midy = int(mesh.nCy/2)
 #==============================================================================
 
 #%% Plot obs data
-PF.Magnetics.plot_obs_2D(rxLoc,d,'Observed Data')
+PF.Magnetics.plot_obs_2D(rxLoc,d, 'Observed Data')
 
 #%% Run inversion
 prob = PF.Magnetics.Problem3D_Integral(mesh, mapping=idenMap, actInd=actv)
@@ -57,7 +51,7 @@ survey.pair(prob)
 #survey.dobs=d
 #survey.mtrue = model
 # Write out the predicted
-pred = prob.fields(mstart)
+pred = prob.fields(driver.m0)
 PF.Magnetics.writeUBCobs('Pred.dat', survey, pred)
 
 wr = np.sum(prob.G**2.,axis=0)**0.5 / mesh.vol[actv]
@@ -72,15 +66,14 @@ plt.xlabel('x');plt.ylabel('z')
 plt.gca().set_aspect('equal', adjustable='box')
 
 reg = Regularization.Simple(mesh, indActive=actv, mapping=idenMap)
-reg.mref = mref
+reg.mref = driver.mref
 reg.wght = wr
 #reg.alpha_s = 1.
 
 # Create pre-conditioner
-diagA = np.sum(prob.G**2.,axis=0) + beta_in*(reg.W.T*reg.W).diagonal()
+diagA  = np.sum(prob.G**2.,axis=0) + beta_in*(reg.W.T*reg.W).diagonal()
 PC     = Utils.sdiag(diagA**-1.)
 
-
 dmis = DataMisfit.l2_DataMisfit(survey)
 dmis.Wd = 1/wd
 opt = Optimization.ProjectedGNCG(maxIter=10,lower=0.,upper=1., maxIterCG= 20, tolCG=1e-3)
@@ -94,10 +87,8 @@ target = Directives.TargetMisfit()
 
 inv = Inversion.BaseInversion(invProb, directiveList=[beta,target])
 
-m0 = mstart
-
 # Run inversion
-mrec = inv.run(m0)
+mrec = inv.run(driver.m0)
 
 m_out = actvMap*mrec
 
@@ -145,7 +136,7 @@ phid =  invProb.phi_d
 
 reg = Regularization.Sparse(mesh, indActive = actv, mapping = idenMap)
 reg.recModel = mrec
-reg.mref = mref
+reg.mref = driver.mref
 reg.wght = wr
 reg.eps_p = eps_p
 reg.eps_q = eps_q
diff --git a/simpegPF/Magnetics.py b/simpegPF/Magnetics.py
index 64fa6363..4665ef06 100644
--- a/simpegPF/Magnetics.py
+++ b/simpegPF/Magnetics.py
@@ -8,9 +8,9 @@ class Problem3D_Integral(Problem.BaseProblem):
 
     #surveyPair = Survey.LinearSurvey
 
-    storeG = True #: Store the forward matrix by default, otherwise just compute d
-    actInd = None #: Active cell indices provided
-    M = None #: Magnetization matrix provided, otherwise all induced
+    forwardOnly = True #: Store the forward matrix by default, otherwise just compute d
+    actInd      = None #: Active cell indices provided
+    M           = None #: Magnetization matrix provided, otherwise all induced
 
 
     def __init__(self, mesh, mapping=None, **kwargs):
diff --git a/simpegPF/MagneticsIO.py b/simpegPF/MagneticsDriver.py
similarity index 100%
rename from simpegPF/MagneticsIO.py
rename to simpegPF/MagneticsDriver.py
diff --git a/simpegPF/Tests/test_magnetics_IO.py b/simpegPF/Tests/test_magnetics_IO.py
index 70b432d4..06d7df41 100644
--- a/simpegPF/Tests/test_magnetics_IO.py
+++ b/simpegPF/Tests/test_magnetics_IO.py
@@ -14,7 +14,7 @@ class MagSensProblemTests(unittest.TestCase):
 
     def test_magnetics_inversion(self):
 
-        driver = PF.MagneticsIO.MagneticsDriver_Inv('assets/magnetics/SimPEG_MAG3D_inv.inp')
+        driver = PF.MagneticsDriver.MagneticsDriver_Inv('assets/magnetics/SimPEG_MAG3D_inv.inp')
 
         print driver.mesh
         print driver.survey
diff --git a/simpegPF/__init__.py b/simpegPF/__init__.py
index 5624fc10..96fafd72 100644
--- a/simpegPF/__init__.py
+++ b/simpegPF/__init__.py
@@ -3,4 +3,4 @@ import BaseMag
 import Magnetics
 import BaseGrav
 import Gravity
-import MagneticsIO
+import MagneticsDriver