Redid how the prompt is generated