Update documentation, update prompt logic