reorganizing

demo.py → demos/demo.py

-from sugar import build_starblade, starblade_iteration
-from matplotlib import pyplot as plt
+import numpy as np
 from astropy.io import fits
+from matplotlib import pyplot as plt
 
-import numpy as np
+from starblade.sugar import build_starblade, starblade_iteration
 
 if __name__ == '__main__':
     #specifying location of the input file:

multichannel_demo.py → demos/multichannel_demo.py

-from sugar import build_multi_starblade, multi_starblade_iteration
-from matplotlib import pyplot as plt
 import numpy as np
+from matplotlib import pyplot as plt
+from starblade.sugar import build_multi_starblade, multi_starblade_iteration
 
 if __name__ == '__main__':
 

sugar.py → starblade/sugar.py

-import nifty4 as ift
-import numpy as np
-from matplotlib import pyplot as plt
 from multiprocessing import Pool
-from starblade_energy import StarbladeEnergy
 
+import nifty4 as ift
+
+from starblade.starblade_energy import StarbladeEnergy
 
 
 def build_starblade(data, alpha=1.5, q=1e-40, cg_iterations=500):