updated size calculation for belief state