Atom distribution should be using the strategy STATES+KPOINTS and not only STATES
At the moment the code is limited to distribute atoms over the states strategy, making k-point only calculations not efficient.
This can be fixed by changing the multicomm slightly and defining the combined strategies, and storing the joint MPI groups in the group
array. This would allow to have a working ion distribution without state parallelization.