Added a variable to the test class