Add logging back