[MINOR] Logging improvements