Removing functin that had been moved elsewhere
Fixed dataset partitioning