Changed the prompt to account for the logic