Updated first test to allow for multiple tests