Fix script. Update docs