[#3] Added some cleanup to restore the environment after performing our tests