Customizing prompt

See issue #3