refactor
fix prompt