Add model comparison to notebook test

main.py

@@ -1,6 +1,7 @@
-from trees.Stree import Stree
 from sklearn.datasets import make_classification
 
+from trees.Stree import Stree
+
 random_state = 1
 X, y = make_classification(n_samples=1500, n_features=3, n_informative=3,
                            n_redundant=0, n_repeated=0, n_classes=2, n_clusters_per_class=2,