Add logging to tests