final refactor for post edit and save as draft :cheers: