ML Training module added