Federated Learning Code