Removed unused tree_rotation
Add some fixes
Add support for batch dataset in dirl