Adding logging to tests