updated for new stim count