Added draft and undraft functions