added logic to close the web driver only on the second run