changed test_suite so you can pass how many to stim as well