Add ability to easily change number of tests