Quick and hacky model of the distribution of energy in atoms