added prompts to ask for the number of tests required