Add draft post support