Add logging option