Modify for PromptWorks