Add logging capabilities