WIP: make dataset transformations work