diff --git a/release-packaging/Classes/FluidUMAP.sc b/release-packaging/Classes/FluidUMAP.sc
index b47aa08..3efc0f1 100644
--- a/release-packaging/Classes/FluidUMAP.sc
+++ b/release-packaging/Classes/FluidUMAP.sc
@@ -6,11 +6,20 @@ FluidUMAP : FluidDataClient {
 			\numNeighbours, numNeighbours,
 			\minDist, minDist,
 			\iterations, iterations,
-			\learnRate, learnRate,
-			\batchSize, batchSize
+			\learnRate, learnRate
 		])
 	}
 
+	fit{|sourceDataSet, action|
+		this.prSendMsg(\fit,
+			[sourceDataSet.asSymbol], action);
+	}
+
+	transform{|sourceDataSet, destDataSet, action|
+		this.prSendMsg(\transform,
+			[sourceDataSet.asSymbol,  destDataSet.asSymbol], action);
+	}
+
 
 	fitTransform{|sourceDataSet, destDataSet, action|
 		this.prSendMsg(\fitTransform,