Adding @AfterMethod to tests.