Add save as draft capability