Updating LogWrapper and adding test cases