Add custom prompt
Readme update