Add logging support