Add logging setup