added draft functionality