Improve prompt
Merge pull request #2 from ianthedeveloper/patch-1

Improve prompt