add test for custom prompt