Slight improvement to cleanup script.