Updated for new prompt