Made the testing script better.