Refactor: modified single_turn and inference codes