diff --git a/base/dataframe_go.go b/base/dataframe_go.go
index 194d64e..067d249 100644
--- a/base/dataframe_go.go
+++ b/base/dataframe_go.go
@@ -10,26 +10,26 @@ import (
 
 // ConvertDataFrameToInstances converts a DataFrame-go dataframe object to Golearn Fixed Data Grid. Allows for compabitibility between dataframe and golearn's ML models.
 // df is the dataframe Object. classAttrIndex is the index of the class Attribute in the data.i
-func ConvertDataFrameToInstances(df *dataframe.DataFrame, classAttrIndex int) base.FixedDataGrid {
+func ConvertDataFrameToInstances(df *dataframe.DataFrame, classAttrIndex int) FixedDataGrid {
 
 	// Creating Attributes based on Datafraem
 	names := df.Names()
-	attrs := make([]base.Attribute, len(names))
+	attrs := make([]Attribute, len(names))
 
-	newInst := base.NewDenseInstances()
+	newInst := NewDenseInstances()
 
 	for i := range names {
 		col := df.Series[i]
 		if reflect.TypeOf(col.Value(0)).Kind() == reflect.String {
-			attrs[i] = new(base.CategoricalAttribute)
+			attrs[i] = new(CategoricalAttribute)
 			attrs[i].SetName(names[i])
 		} else {
-			attrs[i] = base.NewFloatAttribute(names[i])
+			attrs[i] = NewFloatAttribute(names[i])
 		}
 	}
 
 	// Add the attributes
-	newSpecs := make([]base.AttributeSpec, len(attrs))
+	newSpecs := make([]AttributeSpec, len(attrs))
 	for i, a := range attrs {
 
 		newSpecs[i] = newInst.AddAttribute(a)