count() now uses a transform instead of ConstantInputDStream