Add getDraft method