Add logging feature