Added edit functionality for drafts