Redone the testing class for assignment 4