This is code helps user to automatically divide data among clients for federated learning