Modify ui_prompt func