Add draft status to flow