Moved logic for calculating number of samples [skip ci]