Robust Federated Learning in a Heterogeneous Environment