Add logging capability