Add states for draft saving