Refactor prompt