Change apply function behavior

[MarcCote]

I think the apply function should work at the example level instead dataset level.