Add draft state implementation