Added code to remove test div after the test.