Add stage.tau setting
Refactor prompt