nit: `else if (draft) {`