Finished test and added back the for loop