more imports

demos/multichannel_demo.py

 import numpy as np
 from matplotlib import pyplot as plt
-from starblade.sugar import build_multi_starblade, multi_starblade_iteration
+import starblade as sb
 
 if __name__ == '__main__':
 
@@ -10,10 +10,10 @@ if __name__ == '__main__':
     data = data.astype(float)
     data = data.clip(0.0001)
     alpha = 1.25
-    MultiStarblade = build_multi_starblade(data, alpha)
+    MultiStarblade = sb.build_multi_starblade(data, alpha)
 
     for i in range(10):
-        MultiStarblade = multi_starblade_iteration(MultiStarblade, multiprocessing=True)
+        MultiStarblade = sb.multi_starblade_iteration(MultiStarblade, multiprocessing=True)
 
         #plotting a three channel RGB image in each iteration
         diffuse = np.empty_like(data)
@@ -24,5 +24,3 @@ if __name__ == '__main__':
 
         plt.imsave('rgb_diffuse.jpg',diffuse/255.)
         plt.imsave('rgb_point.jpg',point/255.)
-
-

starblade/__init__.py

-from .sugar import build_starblade, starblade_iteration
+from .sugar import (build_starblade, starblade_iteration,
+                    build_multi_starblade, multi_starblade_iteration)